Description
The Datasets & DataLoaders 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) — ToTensor 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 51:
from torchvision.transforms import ToTensor -> from torchvision.transforms import v2
- Line 59:
transform=ToTensor() -> transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
- Line 66:
transform=ToTensor() -> transform=v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
Files
beginner_source/basics/data_tutorial.py
Copied from sekyondaMeta#70
cc @subramen
Description
The Datasets & DataLoaders tutorial uses the deprecated
torchvision.transforms.ToTensor()transform.Changes needed
Deprecated APIs
torchvision.transforms.ToTensordeprecated since torchvision 0.16 (Oct 2023) —ToTensoremits 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/data_tutorial.pyCopied from sekyondaMeta#70
cc @subramen