Skip to content

Commit e666c49

Browse files
Update src/torchjd/autojac/_jac.py
Co-authored-by: Pierre Quinton <pierre.quinton@epfl.ch>
1 parent d7bbfae commit e666c49

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/torchjd/autojac/_jac.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def jac(
6767
<https://github.com/pytorch/pytorch/issues/138422>`_, `when some tensors have
6868
<https://github.com/TorchJD/torchjd/issues/184>`_ ``retains_grad=True`` or `when using an
6969
RNN on CUDA <https://github.com/TorchJD/torchjd/issues/220>`_, for instance. If you
70-
experience issues with ``backward`` try to use ``parallel_chunk_size=1`` to avoid relying on
70+
experience issues with ``jac`` try to use ``parallel_chunk_size=1`` to avoid relying on
7171
``torch.vmap``.
7272
"""
7373

0 commit comments

Comments
 (0)