Unverified Commit 3966f955 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Update deprecation warning to not mention ConvertImageDtype (#7771)

parent 72dcc170
...@@ -16,7 +16,7 @@ class ToTensor(Transform): ...@@ -16,7 +16,7 @@ class ToTensor(Transform):
.. warning:: .. warning::
:class:`v2.ToTensor` is deprecated and will be removed in a future release. :class:`v2.ToTensor` is deprecated and will be removed in a future release.
Please use instead ``transforms.Compose([transforms.ToImageTensor(), transforms.ConvertImageDtype()])``. Please use instead ``v2.Compose([transforms.ToImageTensor(), v2.ToDtype(torch.float32, scale=True)])``.
This transform does not support torchscript. This transform does not support torchscript.
...@@ -40,7 +40,7 @@ class ToTensor(Transform): ...@@ -40,7 +40,7 @@ class ToTensor(Transform):
def __init__(self) -> None: def __init__(self) -> None:
warnings.warn( warnings.warn(
"The transform `ToTensor()` is deprecated and will be removed in a future release. " "The transform `ToTensor()` is deprecated and will be removed in a future release. "
"Instead, please use `transforms.Compose([transforms.ToImageTensor(), transforms.ConvertImageDtype()])`." "Instead, please use `v2.Compose([transforms.ToImageTensor(), v2.ToDtype(torch.float32, scale=True)])`."
) )
super().__init__() super().__init__()
......
...@@ -11,7 +11,7 @@ from torchvision.transforms import functional as _F ...@@ -11,7 +11,7 @@ from torchvision.transforms import functional as _F
def to_tensor(inpt: Any) -> torch.Tensor: def to_tensor(inpt: Any) -> torch.Tensor:
warnings.warn( warnings.warn(
"The function `to_tensor(...)` is deprecated and will be removed in a future release. " "The function `to_tensor(...)` is deprecated and will be removed in a future release. "
"Instead, please use `to_image_tensor(...)` followed by `convert_image_dtype(...)`." "Instead, please use `to_image_tensor(...)` followed by `to_dtype(..., dtype=torch.float32, scale=True)`."
) )
return _F.to_tensor(inpt) return _F.to_tensor(inpt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment