Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "4011273a86e8eaab435d3e0965df79512b3813ca"
Commit 375c9717 authored by Soumith Chintala's avatar Soumith Chintala
Browse files

fix inconsistency in stl10 unlabeled

parent 0613135a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/stl10.py
View file @ 375c9717
...@@ -71,7 +71,7 @@ class STL10(CIFAR10): ...@@ -71,7 +71,7 @@ class STL10(CIFAR10):
elif self.split == 'unlabeled': elif self.split == 'unlabeled':
self.data, _ = self.__loadfile(self.train_list[2][0]) 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': else: # self.split == 'test':
self.data, self.labels = self.__loadfile( self.data, self.labels = self.__loadfile(
self.test_list[0][0], self.test_list[1][0]) self.test_list[0][0], self.test_list[1][0])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment