Jiarui Fang 0aab52301e [hotfix] fix a bug in model data stats tracing (#655)

2022-04-03 21:48:06 +08:00

..

test_grad_handler.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_kernel.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_moe_group.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_moe_zero_init.py

[hotfix] fix a bug in model data stats tracing (#655)

2022-04-03 21:48:06 +08:00

test_moe_zero_model.py

[zero] adapt zero for unsharded paramters (Optimizer part) (#601)

2022-04-01 20:10:47 +08:00

test_moe_zero_optim.py

[hotfix] fix bugs in testing (#659)

2022-04-02 21:58:47 +08:00