Pytorch implementation of a small GAN network for MNIST, FashionMNIST, and USPS datasets.
| Image Size | Generator | Discriminator |
|---|---|---|
| 28x28 | 6,897 | 3,001 |
| 16x16 | 3,729 | 2,473 |
Run commands (also available in scripts.sh):
| Dataset | Run command |
|---|---|
| MNIST | python main.py --dataset mnist |
| Fashion MNIST | python main.py --dataset fashionmnist |
| USPS | python main.py --dataset usps --image_size 16 |
