ColossalAI/tests/test_zero
Hongxin Liu f313babd11
[gemini] support save state dict in shards (#3581)
* [gemini] support state dict shard

* [gemini] add test state dict shard

* [gemini] polish docstr

* [gemini] fix merge

* [gemini] polish code
2023-04-17 17:11:09 +08:00
..
test_gemini [gemini] support save state dict in shards (#3581) 2023-04-17 17:11:09 +08:00
test_legacy [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_low_level [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00