From 81512ae79d4c40530b125ea6e858d204943be82b Mon Sep 17 00:00:00 2001 From: Jonathan Coletti <88168630+JonathanColetti@users.noreply.github.com> Date: Tue, 5 May 2026 16:48:37 -0400 Subject: [PATCH] Update torchvision transforms in quickstart tutorial --- beginner_source/basics/quickstart_tutorial.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/beginner_source/basics/quickstart_tutorial.py b/beginner_source/basics/quickstart_tutorial.py index 5cce8dcfe9a..0cf469f31fd 100644 --- a/beginner_source/basics/quickstart_tutorial.py +++ b/beginner_source/basics/quickstart_tutorial.py @@ -26,7 +26,7 @@ from torch import nn from torch.utils.data import DataLoader from torchvision import datasets -from torchvision.transforms import ToTensor +from torchvision.transforms import v2 ###################################################################### # PyTorch offers domain-specific libraries such as `TorchText `_, @@ -43,7 +43,7 @@ root="data", train=True, download=True, - transform=ToTensor(), + transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]), ) # Download test data from open datasets. @@ -51,7 +51,7 @@ root="data", train=False, download=True, - transform=ToTensor(), + transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]), ) ######################################################################