Skip to content

GitLab

Commit a7935ea6 authored by Kyryl Truskovskyi's avatar Kyryl Truskovskyi Committed by Francisco Massa
Browse files

cast mnist targer to int (#605)

parent e15e81f1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/mnist.py
View file @ a7935ea6
...@@ -67,7 +67,7 @@ class MNIST(data.Dataset): ...@@ -67,7 +67,7 @@ class MNIST(data.Dataset):
Returns: Returns:
tuple: (image, target) where target is index of the target class. 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 # doing this so that it is consistent with all other datasets
# to return a PIL Image # to return a PIL Image
......
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