PR #4588 Universal ckp fixes

Universal ckp fixes #4588

tjruwase merged 7 commits into deepspeedai:master from mosheisland:universal_ckp_fixes

mosheisland requested a review from

jeffra 2 years ago

mosheisland requested a review from

tjruwase 2 years ago

universal-ckp: fix handling of word embedding

90cd14af

mosheisland force pushed from f4152a2f to 2d71d20c 2 years ago

stas00 commented on 2023-10-31

tjruwase commented on 2023-10-31

mosheisland force pushed from 0bac303a to a661958a 2 years ago

universal-ckp: fix optimizer step

a54e8100

universal-ckp: reconfigure model parameter rng tracker

35bd942e

universal-ckp: enforce pattern matching in ds_to_universal

1a227aa9

mosheisland force pushed from 8d154d58 to 1a227aa9 2 years ago

Merge branch 'master' into universal_ckp_fixes

7283ecfe

Merge branch 'master' into universal_ckp_fixes

1e7ca68d

Merge branch 'master' into universal_ckp_fixes

dff18169

tjruwase approved these changes on 2023-11-09

tjruwase merged 8ad187d8 into master 2 years ago

Reviewers

tjruwase

stas00

jeffra

Assignees

No one assigned

Labels

None yet

Milestone

No milestone