DeepSpeed
Fix loading a universal checkpoint
#5263
Merged

Fix loading a universal checkpoint #5263

tjruwase merged 6 commits into master from tohtana/fix_univ_chkpt_load
tohtana
get hp keys from checkpoint files
30993084
tjruwase
tjruwase commented on 2024-03-12
set optimizer's internal states
26b9ea46
resume step in param group
2943e6ab
tohtana Merge branch 'master' into tohtana/fix_univ_chkpt_load
18285aee
tjruwase
tjruwase commented on 2024-03-12
tohtana tohtana marked this pull request as ready for review 1 year ago
tohtana tohtana requested a review from mrwyattii mrwyattii 1 year ago
tohtana tohtana requested a review from awan-10 awan-10 1 year ago
tjruwase
tjruwase approved these changes on 2024-03-13
set step count in opt state for each param
db797112
Merge branch 'tohtana/fix_univ_chkpt_load' of https://github.com/micr…
90c721a9
tjruwase tjruwase merged b112c99e into master 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone