.. |
__init__.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
constants.py
|
f8d3ec7fa1
save/restore step in param groups with zero 1 (#4396)
|
1 year ago |
deepspeed_checkpoint.py
|
198166423d
fix spelling error with deepspeed/ (#3494)
|
1 year ago |
reshape_3d_utils.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
reshape_meg_2d.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
reshape_utils.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
universal_checkpoint.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
utils.py
|
5c3ebd7ede
Clone tensors to avoid torch.save bloat (#3348)
|
1 year ago |
zero_checkpoint.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |