jiamings/wgan - Githubissues

jiamings / wgan

Tensorflow Implementation of Wasserstein GAN (and Improved version in wgan_v2)

238 stars 82 forks source link

generative generative-adversarial-network generative-model tensorflow tensorflow-experiments tensorflow-models

Tensorflow implementation of Wasserstein GAN.

Two versions:

wgan.py: the original clipping method.
wgan_v2.py: the gradient penalty method. (Improved Training of Wasserstein GANs).

How to run (an example):

python wgan_v2.py --data mnist --model mlp --gpus 0