facebookresearch / pytorch_GAN_zoo

A mix of GAN implementations including progressive growing
BSD 3-Clause "New" or "Revised" License
1.61k stars 270 forks source link

How to generate more images? #91

Closed ivder closed 4 years ago

ivder commented 4 years ago

Thanks for the great work. I'm using Cifar-10 dataset and I notice that it will generate 16 images (8x8) like this : image

How to generate more than 8x8 images? Do I need to to modify the source code or the option is available in the config file?

Molugan commented 4 years ago

Hi,

Right now the number of images is hardcoded here https://github.com/facebookresearch/pytorch_GAN_zoo/blob/master/models/trainer/gan_trainer.py#L118

Just beware that if you restart a training model it will reuse the old visualization vector. To avoid that, delete *_refVectors.pt before restarting.