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.