github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 22:43:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
179558a87ab37096450223d8ee4c2b1a06a334a4
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