Skip to content

Commit

Permalink
Update on "[doc] fix amp doc"
Browse files Browse the repository at this point in the history
# Motivation
The doc shown on https://pytorch.org/docs/main/amp.html#torch.amp.autocast_mode.is_autocast_available should be `torch.amp.is_autocast_available` instead of `torch.amp.autocast_mode.is_autocast_available`. This PR aims to fix it.



cc mcarilli ptrblck leslie-fang-intel jgong5

[ghstack-poisoned]
  • Loading branch information
guangyey committed May 11, 2024
2 parents 0865d7e + 1598bcf commit 3d99d26
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion docs/source/amp.rst
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,6 @@ Autocasting
.. currentmodule:: torch.amp

.. autoclass:: DOC_TEST
:members:
.. automethod:: test

.. currentmodule:: torch
Expand Down

0 comments on commit 3d99d26

Please sign in to comment.