Jiarui Fang 193dc8dacb [refactor] refactor the memory utils (#715)		2022-04-11 16:47:57 +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	[refactor] refactor the memory utils (#715)	2022-04-11 16:47:57 +08:00
test_moe_zero_model.py	[zero] improve adaptability for not-shard parameters (#708)	2022-04-11 13:38:51 +08:00
test_moe_zero_optim.py	[refactor] refactor the memory utils (#715)	2022-04-11 16:47:57 +08:00