Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

KeyError : 'D_ema' : error when training with --resume options #6

Open
LeoAlethio opened this issue Nov 16, 2021 · 2 comments
Open

Comments

@LeoAlethio
Copy link

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

I hope that I can get any solution!

Thanks!

@49xxy
Copy link

49xxy commented Jun 13, 2022

Have you solved this problem?

@49xxy
Copy link

49xxy commented Jul 18, 2022

@limbo0000
Hello, I encountered this problem when resume training, and I hope to get your help!
image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants