ver217 a241f61b34 [zero] Update initialize for ZeRO (#458) ... * polish code * shard strategy receive pg in shard() / gather() * update zero engine * polish code		2022-03-18 16:18:31 +08:00
..
common.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_init_context.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_shard_model_v2.py	update sharded optim and fix zero init ctx (#457)	2022-03-18 15:44:47 +08:00
test_shard_param.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_sharded_optim_v2.py	update sharded optim and fix zero init ctx (#457)	2022-03-18 15:44:47 +08:00
test_sharded_optim_with_sync_bn.py	[zero] Update initialize for ZeRO (#458)	2022-03-18 16:18:31 +08:00
test_sharded_optim.py	[unit test] Refactored test cases with component func (#339)	2022-03-11 15:50:28 +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