Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions beginner_source/profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Profiling your PyTorch Module
-----------------------------

**Author:** `Suraj Subramanian <https://github.com/suraj813>`_
**Author:** `Suraj Subramanian <https://github.com/subramen>`_

PyTorch includes a profiler API that is useful to identify the time and
memory costs of various PyTorch operations in your code. Profiler can be
Expand Down Expand Up @@ -316,6 +316,6 @@ def forward(self, input, mask):
# We have seen how Profiler can be used to investigate time and memory bottlenecks in PyTorch models.
# Read more about Profiler here:
#
# - `Profiler Usage Recipe <https://pytorch.org/tutorials/recipes/recipes/profiler.html>`__
# - `Profiler Usage Recipe <https://docs.pytorch.org/tutorials/recipes/recipes/profiler_recipe.html>`__
# - `Profiling RPC-Based Workloads <https://pytorch.org/tutorials/recipes/distributed_rpc_profiling.html>`__
# - `Profiler API Docs <https://pytorch.org/docs/stable/autograd.html?highlight=profiler#profiler>`__