You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Hongxin Liu 50793b35f4 [gemini] accelerate inference (#3641) ... * [gemini] support don't scatter after inference * [chat] update colossalai strategy * [chat] fix opt benchmark * [chat] update opt benchmark * [gemini] optimize inference * [test] add gemini inference test * [chat] fix unit test ci * [chat] fix ci * [chat] fix ci * [chat] skip checkpoint test		2 years ago
..
utils	[gemini] accelerate inference (#3641)	2 years ago
__init__.py	[gemini] accelerate inference (#3641)	2 years ago
albert.py	[Gemini] add albert in test models. (#2075)	2 years ago
beit.py	[testing] add beit model for unit testings (#2196)	2 years ago
bert.py	[test] bert test in non-distributed way (#2074)	2 years ago
gpt2.py	[test] align model name with the file name. (#2045)	2 years ago
hanging_param_model.py	[test] align model name with the file name. (#2045)	2 years ago
inline_op_model.py	…
nested_model.py	…
registry.py	…
repeated_computed_layers.py	[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)	2 years ago
resnet.py	…
simple_net.py	…