szagoruyko · May 6, 2016 19:33
diff --git a/cifar100_pylearn2_torch.lua b/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,
   },
 }
	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,
	},
	}