Skip to content

Commit

Permalink
Update inference_mode doc (pytorch#132321)
Browse files Browse the repository at this point in the history
  • Loading branch information
albanD authored and pytorchmergebot committed Jul 31, 2024
1 parent a488113 commit 4e966e8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/autograd/grad_mode.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ def clone(self) -> "set_grad_enabled":
class inference_mode(_DecoratorContextManager):
r"""Context-manager that enables or disables inference mode.
InferenceMode is a new context manager analogous to :class:`~no_grad`
InferenceMode is a context manager analogous to :class:`~no_grad`
to be used when you are certain your operations will have no interactions
with autograd (e.g., model training). Code run under this mode gets better
performance by disabling view tracking and version counter bumps. Note that
Expand Down

0 comments on commit 4e966e8

Please sign in to comment.