def _init_data(self):
# valset picking:
# for CIFAR, split 5k for val
# for ImageNet, split 3k for val
val_size = 5000 if 'cifar' in data_type else 3000
self.train_loader, self.val_loader, n_class = get_split_dataset(data_type, batch_size,
n_data_worker, val_size,
data_root,
shuffle=False) # same sampling
def _init_data(self):
# split the train set into train + val
# for CIFAR, split 5k for val
# for ImageNet, split 3k for val
val_size = 5000 if 'cifar' in self.data_type else 3000
self.train_loader, self.val_loader, n_class = get_split_dataset(self.data_type, self.batch_size,
self.n_data_worker, val_size,
data_root=self.data_root,
use_real_val=self.use_real_val,
shuffle=False) # same sampling
if self.use_real_val: # use the real val set for eval, which is actually wrong
print('*** USE REAL VALIDATION SET!')