ColossalAI/tests/test_zero
LuGY 839847b7d7
[zero]support zero2 with gradient accumulation (#4511)
* support gradient accumulation with zero2

* fix type
2023-08-25 13:44:07 +08:00
..
test_gemini [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
test_low_level [zero]support zero2 with gradient accumulation (#4511) 2023-08-25 13:44:07 +08:00