https://github.com/emited/VariationalRecurrentNeuralNetwork

Pytorch implementation of the Variational Recurrent Neural Network (VRNN).
https://github.com/emited/VariationalRecurrentNeuralNetwork

Last synced: 2 months ago
JSON representation

Pytorch implementation of the Variational Recurrent Neural Network (VRNN).

• Host: GitHub
• URL: https://github.com/emited/VariationalRecurrentNeuralNetwork
• Owner: emited
• Created: 2017-03-19T15:51:35.000Z (over 7 years ago)
• Default Branch: master
• Last Pushed: 2021-12-13T13:14:40.000Z (over 2 years ago)
• Last Synced: 2024-01-26T20:34:33.376Z (5 months ago)
• Language: Python
• Homepage:
• Size: 12.6 MB
• Stars: 271
• Watchers: 9
• Forks: 70
• Open Issues: 7
• Metadata Files:
  • Readme: README.md

Lists

• Awesome-pytorch-list - VRNN
• Awesome-pytorch-list-CNVersion - VRNN

README

        
# VariationalRecurrentNeuralNetwork

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

The paper is available [here](https://arxiv.org/abs/1506.02216).

![png](images/fig_1_vrnn.png)

## Run:

To train: ``` python train.py ```

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

## Some samples:

![png](images/samples.png)