Skip to content

Conversation

jacobfulano
Copy link
Contributor

In the current mosaicml/composer dev branch, warm up interferes with some learning rate schedulers including cosine annealing warm restarts.

In the following PR, I've changed one line of code in ComposedScheduler that ensures that the cosine annealing warm restarts scheduler only begins after warmup:
self.delay_schedulers = [CosineAnnealingLR, ExponentialLR, CosineAnnealingWarmRestarts]

I've also tested this on a separate branch

@jacobfulano
Copy link
Contributor Author

It seems some of the tests were not passed (despite only changing 2 lines of code). I updated composer to version 0.3, so hopefully this will pass all the tests.

@jacobfulano jacobfulano merged commit f2cf321 into mosaicml:dev Nov 29, 2021
@jacobfulano jacobfulano deleted the warmup_cosine_restarts branch November 29, 2021 19:50
coryMosaicML pushed a commit to coryMosaicML/composer that referenced this pull request Feb 23, 2022
…cml#99)

* warmup and cosine restarts combined nicely

* alphabetical import

* change import order for linting

* test

* Delete experiments directory

Delete experiment yamls

* clean up .gitignore

* run isort
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.

2 participants