HELSON ea13a201bb [polish] polish code for get_static_torch_model (#2405) ...

* [gemini] polish code

* [testing] remove code

* [gemini] make more robust

2023-01-09 17:41:38 +08:00

..

common_utils

…

core

…

model

…

test_colo_checkpoint_tools.py

…

test_comm_spec_apply.py

…

test_context.py

…

test_mix_gather.py

…

test_parameter.py

…

test_shape_consistency_apply.py

…

test_shape_consistency.py

…

test_sharded_linear.py

…

test_sharding_spec.py

…

test_tp_with_zero.py

…