Skip to content
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

fix(train): fix lr_decay not working #291

Merged
merged 1 commit into from
Apr 11, 2023
Merged

Conversation

guranon
Copy link
Contributor

@guranon guranon commented Apr 11, 2023

Since we're using manual optimization, we need to call step() on each of the learning rate schedulers.
This steps learning rates once per epoch so as to match the behavior before the lightning rewrite.
Relevant docs: https://lightning.ai/docs/pytorch/stable/common/optimization.html#learning-rate-scheduling

Closes #292

@guranon
Copy link
Contributor Author

guranon commented Apr 11, 2023

Force push was to change the commit message (TIL about conventional commits)

@34j 34j merged commit 3af223e into voicepaw:main Apr 11, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Learning rate (display) issue
2 participants