We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5efb99a commit 0d23cacCopy full SHA for 0d23cac
1 file changed
beginner_source/basics/optimization_tutorial.py
@@ -28,20 +28,20 @@
28
from torch import nn
29
from torch.utils.data import DataLoader
30
from torchvision import datasets
31
-from torchvision.transforms import ToTensor
+from torchvision.transforms import v2
32
33
training_data = datasets.FashionMNIST(
34
root="data",
35
train=True,
36
download=True,
37
- transform=ToTensor()
+ transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
38
)
39
40
test_data = datasets.FashionMNIST(
41
42
train=False,
43
44
45
46
47
train_dataloader = DataLoader(training_data, batch_size=64)
0 commit comments