ColossalAI/tests/components_to_test
HELSON a3100bd50d
[testing] add beit model for unit testings (#2196)
* [testing] add beit model

* [beit] fix bugs

* [beit] fix bugs

* [testing] fix bugs
2022-12-26 17:35:36 +08:00
..
utils [zero] fix unit-tests (#2039) 2022-11-30 10:40:31 +08:00
__init__.py [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
albert.py [Gemini] add albert in test models. (#2075) 2022-12-05 14:09:34 +08:00
beit.py [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +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