search

pytorch / examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
https://pytorch.org/examples
BSD 3-Clause "New" or "Revised" License
22.21k stars 9.51k forks source link

DCGAN default setting in cifar10 not working #263

Open ToTheBeginning opened 6 years ago

ToTheBeginning commented 6 years ago

After run about 15 epoch, the generator seems to collapse to a parameter setting where it always emits the same point.

Did anyone run the script successfully?

samiede commented 5 years ago

Setting the imageSize from 64 to 32 breaks the convolutional layers, because the kernel sizes don't fit anymore!

rajaswa commented 5 years ago

you can change the kernel size = 2 for final layer of the discriminator and kernel size = 1 for generator. This will work for imagesize = 32