Skip to content

Commit 375c971

Browse files
committed
fix inconsistency in stl10 unlabeled
1 parent 0613135 commit 375c971

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/datasets/stl10.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def __init__(self, root, split='train',
7171

7272
elif self.split == 'unlabeled':
7373
self.data, _ = self.__loadfile(self.train_list[2][0])
74-
self.labels = None
74+
self.labels = np.asarray([-1] * self.data.shape[0])
7575
else: # self.split == 'test':
7676
self.data, self.labels = self.__loadfile(
7777
self.test_list[0][0], self.test_list[1][0])

0 commit comments

Comments
 (0)