cast mnist targer to int (#605)

a7935ea6 · Kyryl Truskovskyi · Francisco Massa · e15e81f1 · a7935ea6
Commit a7935ea6 authored Sep 17, 2018 by Kyryl Truskovskyi Committed by Francisco Massa Sep 17, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

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

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