I have some issues when I try to use my custom dataset.
First. the command that you mentioned on README works for my custom dataset.
python train.py --gpus=8 \
--data=${DATA_PATH} \
--cfg=paper256 \
--outdir=training_example
However, when I try to use pretrained model(ffhq256) and retrain with my custom dataset it gave me some error message.
The command that I tired is
CUDA_VISIBLE_DEVICES=0,1 python3 train.py --gpus=2 --data=${DATA_PATH} --cfg=paper256 --resume=ffhq256 --outdir=training_example
Hi, thanks for your wonderful work.
I have some issues when I try to use my custom dataset.
First. the command that you mentioned on README works for my custom dataset. python train.py --gpus=8 \ --data=${DATA_PATH} \ --cfg=paper256 \ --outdir=training_example
However, when I try to use pretrained model(ffhq256) and retrain with my custom dataset it gave me some error message. The command that I tired is CUDA_VISIBLE_DEVICES=0,1 python3 train.py --gpus=2 --data=${DATA_PATH} --cfg=paper256 --resume=ffhq256 --outdir=training_example
And the error message that I got is this![image](https://user-images.githubusercontent.com/93114598/141927486-6b3a9456-a880-4a59-8f5c-ad72919dab9d.png)
I hope that I can get any solution!
Thanks!