From 873c9146339513b869855c455bd2d755f59d8faa Mon Sep 17 00:00:00 2001
From: Payal  Karnavat <103495865+payal101@users.noreply.github.com>
Date: Tue, 5 May 2026 23:19:57 +0530
Subject: [PATCH] Update trainingyt.py

---
 beginner_source/introyt/trainingyt.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/beginner_source/introyt/trainingyt.py b/beginner_source/introyt/trainingyt.py
index d9f585411e8..5937331e081 100644
--- a/beginner_source/introyt/trainingyt.py
+++ b/beginner_source/introyt/trainingyt.py
@@ -70,11 +70,13 @@
 # PyTorch TensorBoard support
 from torch.utils.tensorboard import SummaryWriter
 from datetime import datetime
-
-
-transform = transforms.Compose(
-    [transforms.ToTensor(),
-    transforms.Normalize((0.5,), (0.5,))])
+# v2 transforms API (torchvision >= 0.18)
+from torchvision.transforms import v2
+transform = v2.Compose([
+   v2.ToImage(),
+   v2.ToDtype(torch.float32, scale=True),
+   v2.Normalize((0.5,), (0.5,))
+])
 
 # Create datasets for training & validation, download if necessary
 training_set = torchvision.datasets.FashionMNIST('./data', train=True, transform=transform, download=True)