We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0613135 commit 375c971Copy full SHA for 375c971
torchvision/datasets/stl10.py
@@ -71,7 +71,7 @@ def __init__(self, root, split='train',
71
72
elif self.split == 'unlabeled':
73
self.data, _ = self.__loadfile(self.train_list[2][0])
74
- self.labels = None
+ self.labels = np.asarray([-1] * self.data.shape[0])
75
else: # self.split == 'test':
76
self.data, self.labels = self.__loadfile(
77
self.test_list[0][0], self.test_list[1][0])
0 commit comments