How does swa work with lr schedulers #11969
Unanswered
maxmatical
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment
-
Yes, When the epoch to use SWA is reached, SWALR (SWA scheduler) will override whatever previous scheduler you had. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Does the lr stay flat once swa callback starts, or does the lr still follow the scheduler? for example if i use a cosine decay schedule over the entire course of training and using the swa callback starting at 0.8 pct of training, will the swa override the lr schedule and keep the lr flat for the last 20% of training?
Beta Was this translation helpful? Give feedback.
All reactions