PyTorch

Improved Training of Wasserstein GANs in pytorch

A pytorch implementation of Paper “Improved Training of Wasserstein GANs”.

Python, NumPy, SciPy, Matplotlib
A recent NVIDIA GPU

A latest master version of Pytorch

  • [x] gan_toy.py : Toy datasets (8 Gaussians, 25 Gaussians, Swiss Roll).(Finished in 2017.5.8)

  • [x] gan_language.py : Character-level language model (Discriminator is using nn.Conv1d. Generator is using nn.Conv1d. Finished in 2017.6.23. Finished in 2017.6.27.)

  • [x] gan_mnist.py : MNIST (Running Results while Finished in 2017.6.26. Discriminator is using nn.Conv1d. Generator is using nn.Conv1d.)

  • [ ] gan_64x64.py: 64×64 architectures(Looking forward to your pull request)

  • [x] gan_cifar.py: CIFAR-10(Great thanks to robotcator)