cifar100_pylearn2_torch.lua

local np = require 'npy4th' local iterm = require 'iterm'

local train = np.loadnpy'./train.npy' local test = np.loadnpy'./test.npy'

local cifar10 = torch.load'/home/zagoruys/raid/datasets/tiny-datasets/cifar10_whitened.t7' print(cifar10)

local train_labels = np.loadnpy'../cifar-100-python/train_labels.npy' local test_labels = np.loadnpy'../cifar-100-python/test_labels.npy'

assert(train:size(1) == train_labels:size(1)) assert(test:size(1) == test_labels:size(1)) print(train_labels:min(), train_labels:max())

local dataset = { trainData = { data = train:view(-1,3,32,32), labels = train_labels:squeeze():double() + 1, size = function() return train_labels:size(1) end, }, testData = { data = test:view(-1,3,32,32), labels = test_labels:squeeze():double() + 1, size = function() return test_labels:size(1) end, }, }