We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 44c2ac2 commit 97bf177Copy full SHA for 97bf177
torchvision/tv_tensors/__init__.py
@@ -10,7 +10,7 @@
10
11
# TODO: Fix this. We skip this method as it leads to
12
# RecursionError: maximum recursion depth exceeded while calling a Python object
13
-# Keeping it here, leads to graph breaks between multiple functional ops instead of having a single graph
+# Until `disable` is removed, there will be graph breaks after all calls to functional transforms
14
@torch.compiler.disable
15
def wrap(wrappee, *, like, **kwargs):
16
"""[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.
0 commit comments