/vae

Variational autoencoders for language modeling / 文本vae模型

Primary LanguagePython

Variational autoencoder for language modeling

Requirement

pytorch>=1.2

usage

To run language modeling on ptb: python run_vae_ptb.py --train --gpu 0

To run language modeling on yahoo: python run_vae_yahoo.py --train --gpu 0

To run language modeling on yelp: python run_vae_yelp.py --train --gpu 0

To test language modeling on ptb: python run_vae_ptb.py --train --gpu 0

To test language modeling on yahoo: python run_vae_yahoo.py --train --gpu 0

To test language modeling on yelp: python run_vae_yelp.py --train --gpu 0