Skip to content

Conversation

KAVYANSHTYAGI
Copy link
Contributor

@KAVYANSHTYAGI KAVYANSHTYAGI commented May 28, 2025

What does this PR do?

This PR adds a prominent documentation warning to the Trainer class docstring (and optionally Profiler docs) to alert users not to wrap Trainer.fit(), Trainer.validate(), or similar methods inside a manual torch.profiler.profile context manager.
Instead, users are advised to use the profiler argument in the Trainer constructor, which is the robust and officially supported method for profiling with Lightning.

This change aims to prevent a common source of cryptic internal errors and crashes reported by users who attempt to manually profile the Trainer, and improves developer experience by making the correct usage obvious in the docs.

Fixes #20779
(You can fill in the relevant issue if one exists, such as #16958)

This PR does not introduce any breaking changes.

Before submitting
Was this discussed/agreed via a GitHub issue?
Yes, this has been discussed in Lightning issues and user forums (see: #16958 and PyTorch #88472).

Did you read the contributor guideline, Pull Request section?

Did you make sure your PR does only one thing, instead of bundling different changes together?

Did you make sure to update the documentation with your changes? (if necessary)

Did you write any new necessary tests? (not for typos and docs)

Did you verify new and existing tests pass locally with your changes?

Did you list all the breaking changes introduced by this pull request?

Did you update the CHANGELOG? (not for typos, docs, test updates, or minor internal changes/refactors)

PR review

Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:

Reviewer checklist

Is this pull request ready for review? (if not, please submit in draft mode)

Check that all items from Before submitting are resolved

Make sure the title is self-explanatory and the description concisely explains the PR

Add labels and milestones (and optionally projects) to the PR so it can be classified
References
PyTorch Issue #88472

Lightning Issue #16958

StackOverflow community discussion

Thanks for considering this! This documentation improvement should help prevent many user headaches and support requests.


📚 Documentation preview 📚: https://pytorch-lightning--20864.org.readthedocs.build/en/20864/

@github-actions github-actions bot added docs Documentation related pl Generic label for PyTorch Lightning package labels May 28, 2025
@KAVYANSHTYAGI KAVYANSHTYAGI requested a review from Borda May 28, 2025 12:35
Copy link
Contributor Author

@KAVYANSHTYAGI KAVYANSHTYAGI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for pointing out the confusion. I’ve removed the unrelated Lightning PR and forum links from the references, now only the directly relevant PyTorch issue is cited to support this documentation warning.

Copy link

stale bot commented Jul 19, 2025

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs. If you need further help see our docs: https://lightning.ai/docs/pytorch/latest/generated/CONTRIBUTING.html#pull-request or ask the assistance of a core contributor here or on Discord. Thank you for your contributions.

@stale stale bot added the won't fix This will not be worked on label Jul 19, 2025
@stale stale bot removed the won't fix This will not be worked on label Aug 14, 2025
@Borda Borda added the waiting on author Waiting on user action, correction, or update label Sep 2, 2025
@Borda Borda changed the title Add documentation warning: Don’t use torch.profiler.profile context manager around Trainer methods docs: warning don’t use torch.profiler.profile context manager around Trainer methods Sep 3, 2025
@Borda Borda merged commit 04e103b into Lightning-AI:master Sep 3, 2025
24 checks passed
Borda added a commit that referenced this pull request Sep 3, 2025
…nd `Trainer` methods (#20864)

* Update profiler.rst
* move documentation to basic

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Nicki Skafte Detlefsen <[email protected]>
Co-authored-by: Jirka Borovec <[email protected]>
(cherry picked from commit 04e103b)
lantiga pushed a commit that referenced this pull request Sep 5, 2025
…nd `Trainer` methods (#20864)

* Update profiler.rst
* move documentation to basic

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Nicki Skafte Detlefsen <[email protected]>
Co-authored-by: Jirka Borovec <[email protected]>
(cherry picked from commit 04e103b)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
docs Documentation related pl Generic label for PyTorch Lightning package waiting on author Waiting on user action, correction, or update
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Pytorch Profiler crashes while using it with Pytorch Lightning modules
3 participants