ver217 62b0a8d644 [zero] sharded optim support hybrid cpu adam (#486) ... * sharded optim support hybrid cpu adam * update unit test * polish docstring		2022-03-22 14:56:59 +08:00
..
common.py	[zero] polish sharded param name (#484)	2022-03-22 14:36:16 +08:00
test_init_context.py	[zero] polish sharded param name (#484)	2022-03-22 14:36:16 +08:00
test_shard_model_v2.py	[test] fixed amp convergence comparison test (#454)	2022-03-18 16:28:16 +08:00
test_shard_param.py	[zero] polish sharded param name (#484)	2022-03-22 14:36:16 +08:00
test_sharded_optim.py	[unit test] Refactored test cases with component func (#339)	2022-03-11 15:50:28 +08:00
test_sharded_optim_v2.py	[zero] sharded optim support hybrid cpu adam (#486)	2022-03-22 14:56:59 +08:00
test_sharded_optim_with_sync_bn.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_state_dict.py	update sharded optim and fix zero init ctx (#457)	2022-03-18 15:44:47 +08:00
test_zero_engine.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_zero_param_mgr.py	using pytest parametrize	2022-03-11 15:50:28 +08:00