stochasticats/WGAN-GP

Implementation of Wasserstein GAN with Gradient Penalty (WGAN-GP)

WGAN-GP

Implementation of Wasserstein GAN with Gradient Penalty (WGAN-GP) in PyTorch. You can read about Wasserstein GAN in the following links:

Wasserstein GAN (Arjovsky, Chintala, and Bottou, 2017): https://arxiv.org/abs/1701.07875

Improved Training of Wasserstein GANs (Gulrajani et al., 2017): https://arxiv.org/abs/1704.00028

MNIST

Testing the model was done on the MNIST dataset: http://yann.lecun.com/exdb/mnist/

Images

(coming soon)