Skip to content

Commit 40b7766

Browse files
SkafteNickiBordajustusschockdeependujha
authored
Add link to pytorch profiler API (#21218)
* add link * Empty-Commit --------- Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: jirka <[email protected]> Co-authored-by: Justus Schock <[email protected]> Co-authored-by: Deependu <[email protected]>
1 parent e5a616f commit 40b7766

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/lightning/pytorch/profilers/pytorch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,8 @@ def __init__(
282282
283283
table_kwargs: Dictionary with keyword arguments for the summary table.
284284
285-
\**profiler_kwargs: Keyword arguments for the PyTorch profiler. This depends on your PyTorch version
285+
\**profiler_kwargs: Keyword arguments for the :class:`Pytorch profiler <torch.profiler.profile>`.
286+
The exact available options depend on your PyTorch version.
286287
287288
Raises:
288289
MisconfigurationException:

0 commit comments

Comments
 (0)