THUzhangga/MovingMNIST

Unsupervised learning of Moving MNIST dataset. This repository contains implemention of ConvLSTM model and PredRNN++ model with Pytorch.

Jupyter Notebook