From ed450b4949d16c74ad9efb85079337dae289e3dc Mon Sep 17 00:00:00 2001 From: Omkar Kabde Date: Tue, 5 May 2026 23:19:50 +0530 Subject: [PATCH] Modernize data tutorial to torchvision v2 API --- beginner_source/basics/data_tutorial.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py index 2c46b33122a..c49f25a587a 100644 --- a/beginner_source/basics/data_tutorial.py +++ b/beginner_source/basics/data_tutorial.py @@ -48,7 +48,7 @@ import torch from torch.utils.data import Dataset from torchvision import datasets -from torchvision.transforms import ToTensor +from torchvision.transforms import v2 import matplotlib.pyplot as plt @@ -56,14 +56,14 @@ root="data", train=True, download=True, - transform=ToTensor() + transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]) ) test_data = datasets.FashionMNIST( root="data", train=False, download=True, - transform=ToTensor() + transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]) )