Skip to content

Add warmup_type arguments in WarmupLR and WarmupDecayLR - #1530

Merged
tjruwase merged 9 commits into
deepspeedai:masterfrom
skpig:warmup_linear_lr
Nov 12, 2021
Merged

tjruwase merged 9 commits into
deepspeedai:masterfrom
skpig:warmup_linear_lr

Conversation

@skpig

@skpig skpig commented Nov 6, 2021

Copy link
Copy Markdown
Contributor

About issue #1298 @tjruwase
I add warmup_type arguments in WarmupLR and WarmupDecayLR. If log is chosen(Default) , the training curve will be the same as before. If linear, then learning rate will increase linearly during warmup.

@ghost

ghost commented Nov 6, 2021

Copy link
Copy Markdown

CLA assistant check
All CLA requirements met.

@tjruwase

tjruwase commented Nov 7, 2021

Copy link
Copy Markdown
Contributor

Thanks. These looks good. Could you please add some unit tests for this here?

@skpig

skpig commented Nov 7, 2021

Copy link
Copy Markdown
Contributor Author

When I ran pytest --forked tests/unit/test_lr_schedulers.py, every test failed with Error reported
AttributeError: Can't pickle local object 'distributed_test.<locals>.dist_wrap.<locals>.dist_init'.
I'm using MacOS with no CUDA. And I already install pytest-forked. Could you please help me deal with it?

@tjruwase

tjruwase commented Nov 7, 2021

Copy link
Copy Markdown
Contributor

@skpig, not a problem for now since the CI will also run the tests. I will investigate any failures.

@skpig

skpig commented Nov 11, 2021

Copy link
Copy Markdown
Contributor Author

Any problems during review? If so, please feel free to at me.

Comment thread deepspeed/runtime/lr_schedules.py Outdated
@tjruwase
tjruwase merged commit 76847f4 into deepspeedai:master Nov 12, 2021
@skpig
skpig deleted the warmup_linear_lr branch November 12, 2021 13:51
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.

3 participants