style-based-gan-pytorch icon indicating copy to clipboard operation
style-based-gan-pytorch copied to clipboard

Bad results when continue training

Open ShaniGam opened this issue 5 years ago • 8 comments

First, thank you for this implementation! I want to continue train a model that I trained before. I run the following command: python train.py --mixing --init_size 64 ./images/ --ckpt ./checkpoint/140000.model I don't get any errors when loading but the generated images don't look like the ones I got before, but instead I get really bad results.

ShaniGam avatar Nov 07 '19 10:11 ShaniGam

Could you use train-step-X.model checkpoint?

rosinality avatar Nov 07 '19 14:11 rosinality

I tried that as well, same results.

ShaniGam avatar Nov 07 '19 14:11 ShaniGam

Then can you check init_size is correct for that checkpoints? I think size will be larger than 64 if you use 140,000 iter checkpoints.

rosinality avatar Nov 07 '19 14:11 rosinality

I'm sure it's the correct size because I stopped the experiment and ran it again right away. Besides, if it generates good images that are > 64, shouldn't generate good 64 images?

ShaniGam avatar Nov 07 '19 14:11 ShaniGam

As generators will be trained for higher resolutions, it will show less quality at lower resolutions.

rosinality avatar Nov 07 '19 23:11 rosinality

Hello, have you solved the problem of continuing training?

chenchenver avatar Dec 07 '19 10:12 chenchenver

Unfortunately no but I stopped working on that.

ShaniGam avatar Dec 17 '19 06:12 ShaniGam

I meet the same problem, i think there might be something wrong when loading the g_optimizer and d_optimizer.

cientgu avatar Apr 24 '20 14:04 cientgu