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,
   },
}