Jiarui Fang 10ef8afdd2 [gemini] init genimi individual directory (#754)		2022-04-14 16:40:26 +08:00
..
common.py	[zero] add tensor placement policies (#743)	2022-04-13 15:00:48 +08:00
test_found_inf.py	[zero] add tensor placement policies (#743)	2022-04-13 15:00:48 +08:00
test_init_context.py	[zero] refactor ShardedParamV2 for convenience (#742)	2022-04-13 14:54:26 +08:00
test_mem_collector.py	[zero] refactor memstats_collector (#746)	2022-04-14 12:01:12 +08:00
test_shard_model_v2.py	[zero] add tensor placement policies (#743)	2022-04-13 15:00:48 +08:00
test_shard_param.py	[zero] refactor ShardedParamV2 for convenience (#742)	2022-04-13 14:54:26 +08:00
test_sharded_optim_v2.py	[zero] add tensor placement policies (#743)	2022-04-13 15:00:48 +08:00
test_sharded_optim_with_sync_bn.py	[utils] correct cpu memory used and capacity in the context of multi-process (#726)	2022-04-12 14:57:54 +08:00
test_state_dict.py	[utils] correct cpu memory used and capacity in the context of multi-process (#726)	2022-04-12 14:57:54 +08:00
test_stateful_tensor_mgr.py	[gemini] init genimi individual directory (#754)	2022-04-14 16:40:26 +08:00
test_tensor_utils.py	[utils] correct cpu memory used and capacity in the context of multi-process (#726)	2022-04-12 14:57:54 +08:00
test_zero_engine.py	[bug] fixed DDP compatibility with torch 1.8 (#739)	2022-04-13 00:08:46 +08:00