fix inconsistency in stl10 unlabeled

375c9717 · Soumith Chintala · 0613135a · 375c9717
Commit 375c9717 authored Aug 22, 2017 by Soumith Chintala
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/stl10.py torchvision/datasets/stl10.py +1 -1

No files found.
--- a/torchvision/datasets/stl10.py
+++ b/torchvision/datasets/stl10.py
@@ -71,7 +71,7 @@ class STL10(CIFAR10):

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