github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-27 02:03:13 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
5a79cffdfd24d6bead2f7c2b674dc69a2ef50129
ColossalAI/tests/test_zero
History
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