Trainer: make checkpoints work under tensor parallelism #48569
Make Trainer checkpoints work under tensor parallelism
ab813aa2
Format
5264b85e
Merge branch 'main' into tp-checkpoint
08ce50cc
Merge branch 'main' into tp-checkpoint
623b3ae5
Merge branch 'main' into tp-checkpoint
4f0364f4
Name the optimizer file in one place
52cbf546
Merge branch 'main' into tp-checkpoint
f837aed4
Merge branch 'main' into tp-checkpoint
02dc1336
Merge branch 'main' into tp-checkpoint
b9b1e958
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub