Skip to content

Instantly share code, notes, and snippets.

@szagoruyko

szagoruyko/make_mnist.lua Secret

Last active May 18, 2016 00:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save szagoruyko/8467ee15d020ab2a7ce80a215af71b74 to your computer and use it in GitHub Desktop.
Save szagoruyko/8467ee15d020ab2a7ce80a215af71b74 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment