SBNet-for-video-prediction

Star-Shape Information Bridge Convolutional LSTM

Environment: Python 3.6.3 Pytorch 1.0

Run download.sh in dataset/ to get the mnist data.

Run main.py to start training

Please pip install progressbar2 before training