ColossalAI/tests/test_moe
HELSON 1a1d68b053
[moe] add checkpoint for moe models (#3354)
* [moe] add checkpoint for moe models

* [hotfix] fix bugs in unit test
2023-03-31 09:20:33 +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_checkpoint.py [moe] add checkpoint for moe models (#3354) 2023-03-31 09:20:33 +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] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
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