Skip to content

Commit 97bf177

Browse files
vfdev-5NicolasHug
andauthored
Update torchvision/tv_tensors/__init__.py
Co-authored-by: Nicolas Hug <nh.nicolas.hug@gmail.com>
1 parent 44c2ac2 commit 97bf177

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/tv_tensors/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
# TODO: Fix this. We skip this method as it leads to
1212
# 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
13+
# Until `disable` is removed, there will be graph breaks after all calls to functional transforms
1414
@torch.compiler.disable
1515
def wrap(wrappee, *, like, **kwargs):
1616
"""[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.

0 commit comments

Comments
 (0)