e-lab / torch-PredNetNew

Torch7 PredNet implementation

This repository is a Torch implementation of the paper Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning. We are looking forward to further improve this network.

Folder description:

1. train: Training the model
2. utils: Contains utility script such as one for preparing dataset for training
3. visualize: Visualize the predicted output alongwith different states of the saved model

First generate dataset compatible for this repository using utils, then train the network, and then use visualization tool to see the trained network's representation.