github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-24 13:03:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
dac127d0ee9a6345295fc76aca0b43f42e22d3ff
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