ColossalAI/tests/test_zero
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
..
test_gemini [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
test_legacy [bf16] add bf16 support (#3882) 2023-06-05 15:58:31 +08:00
test_low_level [zero] add state dict for low level zero (#4179) 2023-07-31 22:13:29 +08:00