Fix torchvision tutorial (#1994)

datumbox · web-flow · commit 47a91e0fa57e · 2022-08-02T08:47:21.000-07:00
diff --git a/intermediate_source/torchvision_tutorial.rst b/intermediate_source/torchvision_tutorial.rst
@@ -329,7 +329,8 @@ transformation:
 
    def get_transform(train):
        transforms = []
-       transforms.append(T.ToTensor())
+       transforms.append(T.PILToTensor())
+       transforms.append(T.ConvertImageDtype(torch.float))
        if train:
            transforms.append(T.RandomHorizontalFlip(0.5))
        return T.Compose(transforms)
