Skip to content

Commit 781b25d

Browse files
committed
Update deprecation warning
1 parent 72dcc17 commit 781b25d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

torchvision/transforms/v2/_deprecated.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class ToTensor(Transform):
4040
def __init__(self) -> None:
4141
warnings.warn(
4242
"The transform `ToTensor()` is deprecated and will be removed in a future release. "
43-
"Instead, please use `transforms.Compose([transforms.ToImageTensor(), transforms.ConvertImageDtype()])`."
43+
"Instead, please use `v2.Compose([transforms.ToImageTensor(), v2.ToDtype(torch.float32, scale=True)])`."
4444
)
4545
super().__init__()
4646

torchvision/transforms/v2/functional/_deprecated.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
def to_tensor(inpt: Any) -> torch.Tensor:
1212
warnings.warn(
1313
"The function `to_tensor(...)` is deprecated and will be removed in a future release. "
14-
"Instead, please use `to_image_tensor(...)` followed by `convert_image_dtype(...)`."
14+
"Instead, please use `to_image_tensor(...)` followed by `to_dtype(..., dtype=torch.float32, scale=True)`."
1515
)
1616
return _F.to_tensor(inpt)
1717

0 commit comments

Comments
 (0)