Updated dataset in torchvision tutorial (#1585)

Fixes issue #759 Co-authored-by: Brian Johnson <brianjo@fb.com>
pytorch · Jun 17, 2021 · 03db2af · 03db2af
1 parent 243acd7
commit 03db2af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intermediate_source/torchvision_tutorial.rst b/intermediate_source/torchvision_tutorial.rst
@@ -112,7 +112,7 @@ Let’s write a ``torch.utils.data.Dataset`` class for this dataset.
    from PIL import Image
 
 
-   class PennFudanDataset(object):
+   class PennFudanDataset(torch.utils.data.Dataset):
        def __init__(self, root, transforms):
            self.root = root
            self.transforms = transforms