Unverified Commit 378a3274 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix error message if v2 transform is not JIT scriptable (#7196)

parent 1120aa9e
......@@ -130,7 +130,7 @@ class Transform(nn.Module):
# is around.
if self._v1_transform_cls is None:
raise RuntimeError(
f"Transform {type(self.__name__)} cannot be JIT scripted. "
f"Transform {type(self).__name__} cannot be JIT scripted. "
f"This is only support for backward compatibility with transforms which already in v1."
f"For torchscript support (on tensors only), you can use the functional API instead."
)
......
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