FeatureLearningRotNet
FeatureLearningRotNet copied to clipboard
CIFAR10 self.data does not have attribute test_labels ,test_data
When run CIFAR10_ConvClassifier_on_RotNet_NIN4blocks_Conv2_feats_K1000.py, there are some errors iin dataloader.py.
if self.dataset_name == 'cifar10':
labels = self.data.test_labels if (self.split == 'test') else self.data.train_labels
data = self.data.test_data if (self.split == 'test') else self.data.train_data
I found that self.data does not have attribute test_labels ,test_data, train_labels, and train_data. How to solve this? Thk.