Jiarui Fang deee317b0f [Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)

2022-12-13 16:34:10 +08:00

..

utils

[zero] fix unit-tests (#2039)

2022-11-30 10:40:31 +08:00

__init__.py

[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)

2022-12-13 16:34:10 +08:00

albert.py

[Gemini] add albert in test models. (#2075)

2022-12-05 14:09:34 +08:00

bert.py

[test] bert test in non-distributed way (#2074)

2022-12-05 13:32:16 +08:00

gpt2.py

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

hanging_param_model.py

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

inline_op_model.py

[testing] fix testing models (#2036)

2022-11-29 13:42:06 +08:00

nested_model.py

[testing] fix testing models (#2036)

2022-11-29 13:42:06 +08:00

registry.py

[test] refactored testing components (#324)

2022-03-11 15:50:28 +08:00

repeated_computed_layers.py

[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)

2022-12-13 16:34:10 +08:00

resnet.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

simple_net.py

[testing] fix testing models (#2036)

2022-11-29 13:42:06 +08:00