ColossalAI/tests/test_checkpoint_io
Hongxin Liu 3c07a2846e
[plugin] a workaround for zero plugins' optimizer checkpoint (#3780)
* [test] refactor torch ddp checkpoint test

* [plugin] update low level zero optim checkpoint

* [plugin] update gemini optim checkpoint
2023-05-19 19:42:31 +08:00
..
test_gemini_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
test_general_checkpoint_io.py [booster] add tests for ddp and low level zero's checkpointio (#3715) 2023-05-10 12:17:02 +08:00
test_low_level_zero_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
test_torch_ddp_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
utils.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00