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 <6035284+Borda@users.noreply.github.com> Co-authored-by: jirka <jirka.borovec@seznam.cz> Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> Co-authored-by: Deependu <deependujha21@gmail.com>
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)