.. |
__init__.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 months ago |
constants.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 months ago |
deepspeed_checkpoint.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
ds_to_universal.py
|
c292b03a40
Improve parallel process of universal checkpoint conversion (#5343)
|
6 months ago |
reshape_3d_utils.py
|
8998707a2f
Universal Checkpoint for Sequence Parallelism (#4752)
|
10 months ago |
reshape_meg_2d.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
reshape_utils.py
|
c946a34220
Fix sort of zero checkpoint files (#5342)
|
6 months ago |
universal_checkpoint.py
|
2c51aba0b7
Add custom reshaping for universal checkpoint (#5390)
|
6 months ago |
utils.py
|
5c3ebd7ede
Clone tensors to avoid torch.save bloat (#3348)
|
1 year ago |
zero_checkpoint.py
|
c56a4b9e0d
Improve universal checkpoint (#5289)
|
6 months ago |