ColossalAI/tests/test_moe
Jiarui Fang 1e885329f4
[test] align model name with the file name. (#2045)
2022-11-30 15:45:26 +08:00
..
test_grad_handler.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
test_kernel.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
test_moe_colo_init.py [moe] initialize MoE groups by ProcessGroup (#1640) 2022-09-23 17:20:41 +08:00
test_moe_group.py
test_moe_zero_init.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
test_moe_zero_model.py [test] align model name with the file name. (#2045) 2022-11-30 15:45:26 +08:00
test_moe_zero_optim.py [test] align model name with the file name. (#2045) 2022-11-30 15:45:26 +08:00