ColossalAI/colossalai/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
..
gemini [gemini] support save state dict in shards (#3581) 2023-04-17 17:11:09 +08:00
legacy [example] update examples related to zero/gemini (#3431) 2023-04-04 17:32:51 +08:00
low_level [misc] add verbose arg for zero and op builder (#3552) 2023-04-17 11:25:35 +08:00
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
wrapper.py [misc] add verbose arg for zero and op builder (#3552) 2023-04-17 11:25:35 +08:00