Use consistent datatypes in DataSet targets (#389)

cbf19d4f · Chang · Francisco Massa · 1dd2d66e · cbf19d4f · cbf19d4f
Commit cbf19d4f authored Jan 30, 2018 by Chang Committed by Francisco Massa Jan 29, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

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

No files found.
--- a/torchvision/datasets/semeion.py
+++ b/torchvision/datasets/semeion.py
@@ -60,7 +60,7 @@ class SEMEION(data.Dataset):
        Returns:
            tuple: (image, target) where target is index of the target class.
        """
-        img, target = self.data[index], self.labels[index]
+        img, target = self.data[index], int(self.labels[index])

        # doing this so that it is consistent with all other datasets
        # to return a PIL Image

--- a/torchvision/datasets/svhn.py
+++ b/torchvision/datasets/svhn.py
@@ -89,7 +89,7 @@ class SVHN(data.Dataset):
        Returns:
            tuple: (image, target) where target is index of the target class.
        """
-        img, target = self.data[index], self.labels[index]
+        img, target = self.data[index], int(self.labels[index])

        # doing this so that it is consistent with all other datasets
        # to return a PIL Image