espectre / pytorch-generative-adversarial-networks

A very simple generative adversarial network (GAN) in PyTorch

Generative Adversarial Networks (GANs) in PyTorch

Introduction

See https://medium.com/@devnag/generative-adversarial-networks-gans-in-50-lines-of-code-pytorch-e81b79659e3f#.sch4xgsa9 for the relevant blog post.

Running

Run the sample code by typing:

./gan_pytorch.py

...and you'll train two nets to battle it out on a shifted/scaled Gaussian distribution. The 'fake' distribution should match the 'real' one within a reasonable time.