Description
The PyTorch Quickstart tutorial uses the deprecated torchvision.transforms.ToTensor() transform.
Changes needed
Deprecated APIs
| Issue |
Current Code |
Replacement |
torchvision.transforms.ToTensor deprecated since torchvision 0.16 (Oct 2023) — emits a deprecation warning; v2 transforms are the supported API |
from torchvision.transforms import ToTensor / transform=ToTensor() |
from torchvision.transforms import v2 / transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]) |
Specific lines
- Line 29:
from torchvision.transforms import ToTensor → from torchvision.transforms import v2
- Line 46:
transform=ToTensor(), → transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),
- Line 54:
transform=ToTensor(), → transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),
Files
beginner_source/basics/quickstart_tutorial.py
Copied from sekyondaMeta#69
cc @subramen
Description
The PyTorch Quickstart tutorial uses the deprecated
torchvision.transforms.ToTensor()transform.Changes needed
Deprecated APIs
torchvision.transforms.ToTensordeprecated since torchvision 0.16 (Oct 2023) — emits a deprecation warning; v2 transforms are the supported APIfrom torchvision.transforms import ToTensor/transform=ToTensor()from torchvision.transforms import v2/transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])Specific lines
from torchvision.transforms import ToTensor→from torchvision.transforms import v2transform=ToTensor(),→transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),transform=ToTensor(),→transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]),Files
beginner_source/basics/quickstart_tutorial.pyCopied from sekyondaMeta#69
cc @subramen