Skip to content

Update Optimization tutorial — replace deprecated ToTensor() with v2 transforms #72

@sekyondaMeta

Description

@sekyondaMeta

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    docathon-2026Docathon 2026easyGood first issue / easy fixintroIntroductory tutorial

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions