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): ...@@ -130,7 +130,7 @@ class Transform(nn.Module):
# is around. # is around.
if self._v1_transform_cls is None: if self._v1_transform_cls is None:
raise RuntimeError( 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"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." 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