ColossalAI/colossalai/zero/low_level
LuGY dd7cc58299 [zero] add state dict for low level zero (#4179)
* add state dict for zero

* fix unit test

* polish
2023-07-31 22:13:29 +08:00
..
bookkeeping [zero] refactor low level zero for shard evenly (#4030) 2023-07-31 22:13:29 +08:00
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
_utils.py [zero] allow passing process group to zero12 (#4153) 2023-07-31 22:13:29 +08:00
low_level_optim.py [zero] add state dict for low level zero (#4179) 2023-07-31 22:13:29 +08:00