You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_legacy/test_moe
hxwang 70c9924d0d
[chore] solve moe ckpt test failure and some other arg pass failure
4 months ago
..
moe_utils.py [moe] clean legacy code 4 months ago
test_grad_handler.py [moe] clean legacy code 4 months ago
test_moe_group.py [moe] clean legacy code 4 months ago
test_moe_hybrid_zero.py [moe] clean legacy code 4 months ago
test_moe_load_balance.py [moe] clean legacy code 4 months ago