Skip to content

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

Description

@LeoAlethio

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions