ColossalAI/tests/test_zero/low_level_zero
Frank Lee 638a07a7f9
[test] fixed gemini plugin test (#3411)
* [test] fixed gemini plugin test

* polish code

* polish code
2023-04-03 17:12:22 +08:00
..
test_grad_acc.py [zero] add zero wrappers (#2523) 2023-01-29 17:52:58 +08:00
test_zero1_2.py [test] fixed gemini plugin test (#3411) 2023-04-03 17:12:22 +08:00
test_zero_init.py [zero] fix gradient clipping in hybrid parallelism (#2521) 2023-01-29 15:09:57 +08:00
test_zero_tp.py [zero] add zero wrappers (#2523) 2023-01-29 17:52:58 +08:00