BigGAN-pytorch icon indicating copy to clipboard operation
BigGAN-pytorch copied to clipboard

Mode collapse with the default parameters

Open hangzhaomit opened this issue 5 years ago • 8 comments

I used the suggested scripts to train on ImageNet and LSUN(2 class). But cannot get the reported results. Most generated images collapse to the same pattern.

hangzhaomit avatar Dec 07 '18 19:12 hangzhaomit

In this project, I use 4 GPUs

sxhxliang avatar Dec 08 '18 09:12 sxhxliang

@AaronLeong why would more GPUs make a difference?

jcpeterson avatar Dec 08 '18 19:12 jcpeterson

@hangzhaomit I assume it means larger batch sizes. 64 requires 4 GPUs and that's still quite small compared to the original paper

jcpeterson avatar Dec 08 '18 19:12 jcpeterson

@AaronLeong I am also using 4 gpus, here is the command I used, can you help to check? Thanks!

python main.py --batch_size 64  --dataset lsun --adv_loss hinge --version biggan_lsun --image_path PATH --parallel True --gpus 0,1,2,3 --use_tensorboard True

hangzhaomit avatar Dec 08 '18 20:12 hangzhaomit

@AaronLeong Also getting mode collapse on another dataset

jcpeterson avatar Dec 09 '18 11:12 jcpeterson

@hangzhaomit Are you using 12GB GPUs or 16GB GPUs?

jcpeterson avatar Dec 11 '18 01:12 jcpeterson

@jcpeterson 12GB, the batch size fits, so it should not matter

hangzhaomit avatar Dec 11 '18 01:12 hangzhaomit

Whats your results on ImageNet?I‘m trainning on the ImageNet(only two classes) and get bad results.

imsyu avatar Dec 29 '18 09:12 imsyu