ColossalAI/colossalai/zero/low_level/bookkeeping
LuGY 839847b7d7
[zero]support zero2 with gradient accumulation (#4511)
* support gradient accumulation with zero2

* fix type
2023-08-25 13:44:07 +08:00
..
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
base_store.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
bucket_store.py [hotfix] fix unsafe async comm in zero (#4404) 2023-08-11 15:09:24 +08:00
gradient_store.py [zero]support zero2 with gradient accumulation (#4511) 2023-08-25 13:44:07 +08:00
parameter_store.py [zero] refactor low level zero for shard evenly (#4030) 2023-07-31 22:13:29 +08:00
tensor_bucket.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00