Skip to content

Latest commit

 

History

5 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

WGAN in Tensorflow

Implementation of a WGAN for MNIST, with checkpoint in model/

Requirements

  • tensorflow=1.0.0
  • numpy
  • tqdm
  • argparse

Train

To train the model, run the command:

python model.py --train

Model retrieve automaticcaly the last checkpoint and restart at the global step, and saved itself every 100 iterations.
It also saved summary for images generated and other variables, to further display in Tensorboard tensorboard --logdir="logs"

Draw

To draw sample from the generator, run the command:

python model.py --draw 1,2,3,4,5,7,9

Samples generated

Alt text

About

Wassertein GAN in Tensorflow

Topics

Resources

Stars

4 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages