ColossalAI/tests/test_checkpoint_io
Hongxin Liu 5452df63c5
[plugin] torch ddp plugin supports sharded model checkpoint (#3775)
* [plugin] torch ddp plugin add save sharded model

* [test] fix torch ddp ckpt io test

* [test] fix torch ddp ckpt io test

* [test] fix low level zero plugin test

* [test] fix low level zero plugin test

* [test] add debug info

* [test] add debug info

* [test] add debug info

* [test] add debug info

* [test] add debug info

* [test] fix low level zero plugin test

* [test] fix low level zero plugin test

* [test] remove debug info
2023-05-18 20:05:59 +08:00
..
test_gemini_checkpoint_io.py [booster] add tests for ddp and low level zero's checkpointio (#3715) 2023-05-10 12:17:02 +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 [devops] update torch version of CI (#3725) 2023-05-15 17:20:56 +08:00
test_torch_ddp_checkpoint_io.py [plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2023-05-18 20:05:59 +08:00