From 7a2043f9cbdf30ade4886ab111821d8cebcda9ac Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Fri, 5 Feb 2021 11:26:50 +0100 Subject: [PATCH 1/2] fix STL10 folds --- torchvision/datasets/stl10.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/datasets/stl10.py b/torchvision/datasets/stl10.py index 35c71bd2cf1..6db5714827b 100644 --- a/torchvision/datasets/stl10.py +++ b/torchvision/datasets/stl10.py @@ -181,7 +181,7 @@ def __load_folds(self, folds: Optional[int]) -> None: self.root, self.base_folder, self.folds_list_file) with open(path_to_folds, 'r') as f: str_idx = f.read().splitlines()[folds] - list_idx = np.fromstring(str_idx, dtype=np.uint8, sep=' ') + list_idx = np.fromstring(str_idx, dtype=np.long, sep=' ') self.data = self.data[list_idx, :, :, :] if self.labels is not None: self.labels = self.labels[list_idx] From bb954914d85643956d7bd38026bed27a14f8df06 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Fri, 5 Feb 2021 11:44:52 +0100 Subject: [PATCH 2/2] use np.int64 over np.long --- torchvision/datasets/stl10.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/datasets/stl10.py b/torchvision/datasets/stl10.py index 6db5714827b..50e9af882bc 100644 --- a/torchvision/datasets/stl10.py +++ b/torchvision/datasets/stl10.py @@ -181,7 +181,7 @@ def __load_folds(self, folds: Optional[int]) -> None: self.root, self.base_folder, self.folds_list_file) with open(path_to_folds, 'r') as f: str_idx = f.read().splitlines()[folds] - list_idx = np.fromstring(str_idx, dtype=np.long, sep=' ') + list_idx = np.fromstring(str_idx, dtype=np.int64, sep=' ') self.data = self.data[list_idx, :, :, :] if self.labels is not None: self.labels = self.labels[list_idx]