github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-14 12:10:55 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
12eff9eb4cb74e0c20525aa30dc64375832afe45
ColossalAI/colossalai/zero
History
Hongxin Liu 12eff9eb4c [gemini] state dict supports fp16 (#3590) 
* [gemini] save state dict support fp16

* [gemini] save state dict shard support fp16

* [gemini] fix state dict

* [gemini] fix state dict
2023-04-19 11:01:48 +08:00
..
gemini
[gemini] state dict supports fp16 (#3590)
2023-04-19 11:01:48 +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