stylegan2-pytorch

stylegan2-pytorch copied to clipboard

I want to use prepare_data.py

python prepare_data.py --out LMDB_PATH --n_worker N_WORKER --size SIZE1,SIZE2,SIZE3,... DATASET_PATH

However, when I was trying to use that code, (my data is in folder "numbered_SNUH_knee_512")

python prepare_data.py ../numbered_SNUH_knee_512

error has been occurred.

[File "prepare_data.py", line 104, in <module>
    with lmdb.open(args.out, map_size=1024 ** 4, readahead=False) as env:
TypeError: 'path' argument required]

How am I supposed to write properly?

Thank you

You need to set --out arguments.

Aha! Thank you so much for the answer.

I have another question actually,

python -m torch.distributed.launch --nproc_per_node=N_GPU --master_port=PORT train.py --batch BATCH_SIZE LMDB_PATH in this code, what is the PORT? I cannot figure out what that means in this context.

Thank you

It is arbitrary ports that will be used for distributed training. You can use some free ports.

hello, does it take only images, not labels? Thanks