HELSON e5d615aeee [hotfix] fix bugs in testing (#659) ... * remove hybrid adam in test_moe_zero_optim * fix activation checkpointing and its unitest		2022-04-02 21:58:47 +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	[zero] adapt zero for unsharded paramters (Optimizer part) (#601)	2022-04-01 20:10:47 +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