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.
Run: qlua main.lua -useGPU -dataBig -nlayers 3 -display -save -savePics
to train on GPU on large dataset, visualize and save model and results.
- Memory usage: decrease
nSeq
parameter to use less memory. The model is clonednSeq
times and therefore memory usage is proportional to it.