ColossalAI/tests/kit
Bin Jia 5c6f183192 [test] Hotfix/fix some model test and refactor check util api (#4369)
* fix llama test

* fix test bug of bert, blip2, bloom, gpt2

* fix llama test

* fix opt test

* fix sam test

* fix sam test

* fix t5 test

* fix vit test

* fix whisper test

* fix whisper test

* polish code

* adjust allclose parameter

* Add mistakenly deleted code

* addjust allclose

* change loss function for some base model
2023-08-15 23:25:14 +08:00
..
model_zoo [test] Hotfix/fix some model test and refactor check util api (#4369) 2023-08-15 23:25:14 +08:00
__init__.py [test] added timm models to test model zoo (#3129) 2023-03-14 14:29:18 +08:00