Skip to content

A implementation of SeqGAN in PyTorch, following the implementation in tensorflow.

Notifications You must be signed in to change notification settings

ZiJianZhao/SeqGAN-PyTorch

Repository files navigation

SeqGAN-PyTorch

A implementation of SeqGAN in PyTorch, following the implementation in tensorflow.

Tested with:

  • PyTorch v1 Stable
  • Python 3.6
  • CUDA at least 8.0 (For GPU)

Origin

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

Runing

$ python main.py

After runing this file, the results will be printed on terminal. You can change the parameters in the main.py.

About

A implementation of SeqGAN in PyTorch, following the implementation in tensorflow.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages