Jiarui Fang f7e276fa71 [Gemini] add GeminiAdamOptimizer (#1960)		2022-11-16 14:44:28 +08:00
..
common_utils	…
core	…
model	…
test_colo_checkpoint_tools.py	…
test_comm_spec_apply.py	…
test_context.py	[ColoTensor] reconfig ColoInitContext, decouple default_pg and default_dist_spec. (#1953)	2022-11-15 16:24:16 +08:00
test_parameter.py	…
test_shape_consistency.py	…
test_shape_consistency_apply.py	…
test_sharded_linear.py	[ColoTensor] ColoInitContext initialize parameters in shard mode. (#1937)	2022-11-14 16:05:09 +08:00
test_sharding_spec.py	…
test_tp_with_zero.py	[Gemini] add GeminiAdamOptimizer (#1960)	2022-11-16 14:44:28 +08:00