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

Stepwise LR scheduler #20211

Open
wants to merge 38 commits into
base: master
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
5dba6f9
Fix DDP strategy registration with override
01AbhiSingh Jul 23, 2024
3d8b2bf
added ddp alias strategy in strategies/ddp.py
01AbhiSingh Jul 24, 2024
7a55c5c
added ddp alias strategy in strategies/ddp.py
01AbhiSingh Jul 24, 2024
f4b01e5
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 24, 2024
4424d70
Merge branch 'master' into ddp-strategy-alias
01AbhiSingh Jul 27, 2024
607363e
updated tests
01AbhiSingh Aug 6, 2024
3099586
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Aug 6, 2024
935a9c1
updated test_registry.py
01AbhiSingh Aug 6, 2024
c70ef61
Merge branch 'master' into ddp-strategy-alias
01AbhiSingh Aug 6, 2024
ebfedf6
updated test_cli.py
01AbhiSingh Aug 6, 2024
3285d7a
Merge branch 'ddp-strategy-alias' of https://github.com/01AbhiSingh/p…
01AbhiSingh Aug 6, 2024
4b7b719
Stepwise LR scheduler not working across epochs
01AbhiSingh Aug 16, 2024
5be642f
Merge remote-tracking branch 'origin' into Stepwise-LR-scheduler
01AbhiSingh Aug 16, 2024
fc01630
Merge branch 'master' into stepwiseLRscheduler
01AbhiSingh Aug 21, 2024
7f748cf
Merge branch 'master' into stepwiseLRscheduler
01AbhiSingh Aug 28, 2024
06f0a0a
Merge branch 'master' into stepwiseLRscheduler
01AbhiSingh Sep 9, 2024
3c48c9e
Merge branch 'master' into stepwiseLRscheduler
Borda Sep 27, 2024
63cd1f0
Added test for LR scheduler stepping across epoch boundaries
01AbhiSingh Dec 7, 2024
48a7c8e
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 7, 2024
64ed819
added the required changes
01AbhiSingh Dec 11, 2024
29af194
added the required changes
01AbhiSingh Dec 11, 2024
09bc52b
added the required changes
01AbhiSingh Dec 11, 2024
e96c474
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 11, 2024
2391336
added the required changes
01AbhiSingh Dec 12, 2024
a273722
Merge branch 'stepwiseLRscheduler' of https://github.com/01AbhiSingh/…
01AbhiSingh Dec 12, 2024
eb98dce
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 12, 2024
e45a8f9
added the dataloader function and added the following lib from torch.…
01AbhiSingh Dec 12, 2024
7adad14
Merge branch 'stepwiseLRscheduler' of https://github.com/01AbhiSingh/…
01AbhiSingh Dec 12, 2024
7bb9697
Merge branch 'master' into stepwiseLRscheduler
01AbhiSingh Dec 12, 2024
4c77cb3
Merge branch 'master' into stepwiseLRscheduler
01AbhiSingh Dec 12, 2024
15052fb
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 12, 2024
e30a504
added the changes
01AbhiSingh Dec 13, 2024
9dbbc8d
Merge branch 'stepwiseLRscheduler' of https://github.com/01AbhiSingh/…
01AbhiSingh Dec 13, 2024
27047bf
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Dec 13, 2024
ac5afed
added the changes
01AbhiSingh Feb 3, 2025
c61fd46
added the changes
01AbhiSingh Feb 3, 2025
e17fd6f
Merge branch 'master' into stepwiseLRscheduler
lantiga Feb 3, 2025
337c1c2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Feb 3, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
updated test_registry.py
  • Loading branch information
01AbhiSingh committed Aug 6, 2024
commit 935a9c1a4ae8a003f66955e22f21f627b46d9f6b
1 change: 1 addition & 0 deletions tests/tests_fabric/strategies/test_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ def __init__(self, param1, param2):
def test_available_strategies_in_registry():
expected = {
"ddp",
"ddp_find_unused_parameters_true",
"deepspeed",
"deepspeed_stage_1",
"deepspeed_stage_1_offload",
Expand Down
Loading