From 16626792f31ad5e3b29427abff0450f982de3c10 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 28 Jul 2023 10:31:14 +0100 Subject: [PATCH] update docstring --- torchvision/transforms/v2/_deprecated.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/v2/_deprecated.py b/torchvision/transforms/v2/_deprecated.py index cbc5bd5b915..1cb135a3062 100644 --- a/torchvision/transforms/v2/_deprecated.py +++ b/torchvision/transforms/v2/_deprecated.py @@ -16,7 +16,7 @@ class ToTensor(Transform): .. warning:: :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.