You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
I hope that I can get any solution!
Thanks!
The text was updated successfully, but these errors were encountered:
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
I hope that I can get any solution!
Thanks!
The text was updated successfully, but these errors were encountered: