Pytorch implementation of a small GAN network for MNIST, FashionMNIST and USPS dataset.
| Image Size | Generator | Disriminator |
|---|---|---|
| 28x28 | 6,897 | 3,001 |
| 16x16 | 3,729 | 2,473 |
Change the DB variable to change the dataset.
For using the saved model to generate images, set LOAD_MODEL to True and EPOCHS to 0.
