Imported from https://github.com/Ksuryateja/DCGAN-MNIST-pytorch