Pytorch implementation of the Variational RNN (VRNN), from A Recurrent Latent Variable Model for Sequential Data.
The paper is available here.
To train: python train.py
To sample with saved model: python sample.py [saves/saved_state_dict_name.pth]