emited / VariationalRecurrentNeuralNetwork

Pytorch implementation of the Variational Recurrent Neural Network (VRNN).
282 stars 69 forks source link

VariationalRecurrentNeuralNetwork

Pytorch implementation of the Variational RNN (VRNN), from A Recurrent Latent Variable Model for Sequential Data.

The paper is available here.

png

Run:

To train: python train.py

To sample with saved model: python sample.py [saves/saved_state_dict_name.pth]

Some samples:

png