ColossalAI/tests/kit/model_zoo/transformers
Frank Lee ed19290560
[booster] implemented mixed precision class (#3151)
* [booster] implemented mixed precision class

* polish code
2023-03-17 11:00:15 +08:00
..
__init__.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00
albert.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00
bert.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00
gpt.py [booster] implemented mixed precision class (#3151) 2023-03-17 11:00:15 +08:00
opt.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00
t5.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00