mirror of https://github.com/hpcaitech/ColossalAI
3c07a2846e
* [test] refactor torch ddp checkpoint test * [plugin] update low level zero optim checkpoint * [plugin] update gemini optim checkpoint |
||
---|---|---|
.. | ||
test_gemini_checkpoint_io.py | ||
test_general_checkpoint_io.py | ||
test_low_level_zero_checkpoint_io.py | ||
test_torch_ddp_checkpoint_io.py | ||
utils.py |