Skip to content

Commit 1d0706d

Browse files
Merge branch 'main' into fix/optimization-tutorial-v2-transforms
2 parents 0d23cac + 68ed483 commit 1d0706d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

beginner_source/basics/quickstart_tutorial.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from torch import nn
2727
from torch.utils.data import DataLoader
2828
from torchvision import datasets
29-
from torchvision.transforms import ToTensor
29+
from torchvision.transforms import v2
3030

3131
######################################################################
3232
# PyTorch offers domain-specific libraries such as `TorchText <https://pytorch.org/text/stable/index.html>`_,
@@ -43,15 +43,15 @@
4343
root="data",
4444
train=True,
4545
download=True,
46-
transform=ToTensor(),
46+
transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),
4747
)
4848

4949
# Download test data from open datasets.
5050
test_data = datasets.FashionMNIST(
5151
root="data",
5252
train=False,
5353
download=True,
54-
transform=ToTensor(),
54+
transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),
5555
)
5656

5757
######################################################################

0 commit comments

Comments
 (0)