Skip to content

Commit ed450b4

Browse files
committed
Modernize data tutorial to torchvision v2 API
1 parent 5efb99a commit ed450b4

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

beginner_source/basics/data_tutorial.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,22 +48,22 @@
4848
import torch
4949
from torch.utils.data import Dataset
5050
from torchvision import datasets
51-
from torchvision.transforms import ToTensor
51+
from torchvision.transforms import v2
5252
import matplotlib.pyplot as plt
5353

5454

5555
training_data = datasets.FashionMNIST(
5656
root="data",
5757
train=True,
5858
download=True,
59-
transform=ToTensor()
59+
transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
6060
)
6161

6262
test_data = datasets.FashionMNIST(
6363
root="data",
6464
train=False,
6565
download=True,
66-
transform=ToTensor()
66+
transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
6767
)
6868

6969

0 commit comments

Comments
 (0)