Pytorch-Vanilla-GAN

Pytorch implementation of Vanilla-GAN for MNIST, FashionMNIST and USPS dataset.

Generator

FC(10)→FC(1024)→FC(1024)→FC(Image_Size)

Discriminator

FC(Image_Size)→FC(256)→FC(256)→FC(1)

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.

Generated Samples

MNIST

FashionMNIST

USPS