diff --git a/src/lightning/pytorch/profilers/pytorch.py b/src/lightning/pytorch/profilers/pytorch.py index e264d5154feba..7e17dcd40398a 100644 --- a/src/lightning/pytorch/profilers/pytorch.py +++ b/src/lightning/pytorch/profilers/pytorch.py @@ -282,7 +282,8 @@ def __init__( table_kwargs: Dictionary with keyword arguments for the summary table. - \**profiler_kwargs: Keyword arguments for the PyTorch profiler. This depends on your PyTorch version + \**profiler_kwargs: Keyword arguments for the :class:`Pytorch profiler `. + The exact available options depend on your PyTorch version. Raises: MisconfigurationException: