pytorch_proj

Vanilla GAN on MNIST

This project implements the Vanilla GAN(Generative Adversarial Network) on the MNIST dataset. The goal of this project is to generate fake digit images which can be used for data augmentation.

Content

• The step-by-step notebook

• Python script

  Usage: python gan_mnist.py

Dependeicies

• numpy
• pytorch
• torchvision
• tqdm

Results