Description
The Optimizing Model Parameters tutorial uses the deprecated torchvision.transforms.ToTensor() transform.
Changes needed
Deprecated APIs
| Issue |
Current Code |
Replacement |
Since |
torchvision.transforms.ToTensor |
from torchvision.transforms import ToTensor / transform=ToTensor() (lines 31, 34-45) |
from torchvision.transforms import v2 / transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)]) |
torchvision 0.16 (Oct 2023) — ToTensor emits a deprecation warning; v2 transforms are the supported API |
Specific lines
- Line 31:
from torchvision.transforms import ToTensor -> from torchvision.transforms import v2
- Line 37:
transform=ToTensor() -> transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
- Line 44:
transform=ToTensor() -> transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
Files
beginner_source/basics/optimization_tutorial.py
Copied from sekyondaMeta#72
cc @subramen
Description
The Optimizing Model Parameters tutorial uses the deprecated
torchvision.transforms.ToTensor()transform.Changes needed
Deprecated APIs
torchvision.transforms.ToTensorfrom torchvision.transforms import ToTensor/transform=ToTensor()(lines 31, 34-45)from torchvision.transforms import v2/transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])ToTensoremits a deprecation warning; v2 transforms are the supported APISpecific 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/optimization_tutorial.pyCopied from sekyondaMeta#72
cc @subramen