ColossalAI/tests/test_zero/low_level_zero
HELSON d565a24849
[zero] add unit testings for hybrid parallelism (#2486)
2023-01-18 10:36:10 +08:00
..
test_grad_acc.py [zero] polish low level optimizer (#2473) 2023-01-13 14:56:17 +08:00
test_zero1_2.py [zero] polish low level optimizer (#2473) 2023-01-13 14:56:17 +08:00
test_zero_init.py [zero] add unit test for low-level zero init (#2474) 2023-01-15 10:42:01 +08:00
test_zero_tp.py [zero] add unit testings for hybrid parallelism (#2486) 2023-01-18 10:36:10 +08:00