/torch-prednet

Primary LanguageLua

PredNet in Torch7

A model of PredNet in Torch7

sample

get started:

Download dataset: moving MNIST examples. Small is a 100-sample test, otherwise use the larger ones with 8000 samples. This dataset originated from Viorica Patraucean and team.

to train PredNet:

Run: qlua main.lua -useGPU -dataBig -nlayers 3 -display -save -savePics to train on GPU on large dataset, visualize and save model and results.

Notes:

  • Memory usage: decrease nSeq parameter to use less memory. The model is cloned nSeq times and therefore memory usage is proportional to it.