make_mnist.lua

local mnist = require 'mnist'

local train = mnist.traindataset() local test = mnist.testdataset()

local dataset = { trainData = { data = train.data:double():view(train.label:numel(),1,28,28) / 256, labels = train.label:double() + 1, }, testData = { data = test.data:double():view(test.label:numel(),1,28,28) / 256, labels = test.label:double() + 1, }, }

torch.save('mnist.t7', dataset)