From 219d9fd4aa91deced52f840817d860b087bdc628 Mon Sep 17 00:00:00 2001 From: Ryan Szeto Date: Thu, 17 Aug 2017 08:20:50 -0400 Subject: [PATCH] Fixed parallel data loading for Moving MNIST --- data/moving_mnist.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/moving_mnist.lua b/data/moving_mnist.lua index 714b980..2c04829 100644 --- a/data/moving_mnist.lua +++ b/data/moving_mnist.lua @@ -147,7 +147,7 @@ function MovingMNISTDataset:plot() self:plotSeq(savedir .. '/seq.png') end -trainLoader = MovingMNISTLoader(opt, 'train') +trainLoader = MovingMNISTLoader(opt_t or opt, 'train') trainLoader:normalize() -valLoader = MovingMNISTLoader(opt, 'val') +valLoader = MovingMNISTLoader(opt_t or opt, 'val') valLoader:normalize()