mukul09/SeqGAN-Pytorch

An optimized version of SeqGAN in pytorch

SeqGAN Pytorch

An optimized version of Pytorch SeqGan based on the paper SeqGAN: Sequence Generative Adversarial Nets with Policy Gradient and ZijianZhao implementation.

Dependency

Pytorch (0.3.1)

Python 2.7.12

CUDA

Getting Started

python main.py --lr=1e-3 --gpus=0

You may also type

python main.py --help

to see all of the input arguments

Author

Haibin Yu @HeroKillerEver