Pytorch implementation of Vanilla-GAN for MNIST, FashionMNIST and USPS dataset.
FC(10)→FC(1024)→FC(1024)→FC(Image_Size)
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.