Open hangzhaomit opened 5 years ago
In this project, I use 4 GPUs
@AaronLeong why would more GPUs make a difference?
@hangzhaomit I assume it means larger batch sizes. 64 requires 4 GPUs and that's still quite small compared to the original paper
@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
@AaronLeong Also getting mode collapse on another dataset
@hangzhaomit Are you using 12GB GPUs or 16GB GPUs?
@jcpeterson 12GB, the batch size fits, so it should not matter
Whats your results on ImageNet?I‘m trainning on the ImageNet(only two classes) and get bad results.
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.