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