tree: ece6aa90a7db0e802eb978e79ccae86514d049f8 [path history] [tgz]
  1. model/
  2. README.md
  3. download_mnist.py
  4. lsgan.py
  5. utils.py
  6. vanilla.py
examples/gan/README.md

Train a Generative Adversarial Nets (GAN) model

This example is to train a Generative Adversarial Nets (GAN) model over the MNIST dataset.

Running instructions

  1. Download the pre-processed MNIST dataset

  2. Start the training

     python vanilla.py mnist.pkl.gz
    

By default the training code would run on CPU. To run it on a GPU card, please start the program with an additional argument

    python vanilla.py mnist.pkl.gz --use_gpu