A implementation of SeqGAN in PyTorch, following the implementation in tensorflow.
- PyTorch v1 Stable
- Python 3.6
- CUDA at least 8.0 (For GPU)
The idea is from paper SeqGAN: Sequence Generative Adversarial Nets with Policy Gradient
The code is rewrited in PyTorch with the structure largely from Tensorflow Implementation
$ python main.py
After runing this file, the results will be printed on terminal. You can change the parameters in the main.py.