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.
ColossalAI/tests/components_to_test
HELSON a3100bd50d
[testing] add beit model for unit testings (#2196)
2 years ago
..
utils [zero] fix unit-tests (#2039) 2 years ago
__init__.py [testing] add beit model for unit testings (#2196) 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 [testing] fix testing models (#2036) 2 years ago
nested_model.py [testing] fix testing models (#2036) 2 years ago
registry.py [test] refactored testing components (#324) 3 years ago
repeated_computed_layers.py [Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127) 2 years ago
resnet.py [zero] new interface for ShardedOptimv2 (#406) 3 years ago
simple_net.py [testing] fix testing models (#2036) 2 years ago