-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Update auto-opt docs #6037
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update auto-opt docs #6037
Conversation
Codecov Report
@@ Coverage Diff @@
## master #6037 +/- ##
======================================
Coverage 93% 93%
======================================
Files 160 160
Lines 11343 11400 +57
======================================
+ Hits 10554 10621 +67
+ Misses 789 779 -10 |
| def __init__(self): | ||
| self.automatic_optimization = False | ||
| def training_step(self, batch, batch_idx, optimizer_idx): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we should probably also make it so that the opt idx is not even part of the signature when doing manual opt, right? or perhaps it is already like this and just outdated docs.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it's not like that in the code too. opt_idx is a part of the signature and agreed it shouldn't be in case of manual_optimization. @tchaton wdys?
Co-authored-by: Nicki Skafte <[email protected]>
SkafteNicki
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Co-authored-by: Nicki Skafte <[email protected]>
Co-authored-by: Carlos Mocholí <[email protected]>
|
Please, do not merge until all comments are resolved :) |
Co-authored-by: chaton <[email protected]>
What does this PR do?
Updated the docs for automatic_optimization. It's a LightningModule property now. Added docs for a few more properties. Also, update the automatic_optimization code wherever it's required. Other whitespace changes are made by pre-commit.
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
Before you start reviewing make sure you have read Review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃