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
__init__.py
albert.py
beit.py
bert.py
gpt2.py
hanging_param_model.py
inline_op_model.py
nested_model.py
registry.py
repeated_computed_layers.py
resnet.py
simple_net.py