ColossalAI/tests/kit/model_zoo
ver217 6ae8ed0407
[lazyinit] add correctness verification (#3147)
* [lazyinit] fix shared module

* [tests] add lazy init test utils

* [tests] add torchvision for lazy init

* [lazyinit] fix pre op fn

* [lazyinit] handle legacy constructor

* [tests] refactor lazy init test models

* [tests] refactor lazy init test utils

* [lazyinit] fix ops don't support meta

* [tests] lazy init test timm models

* [lazyinit] fix set data

* [lazyinit] handle apex layers

* [tests] lazy init test transformers models

* [tests] lazy init test torchaudio models

* [lazyinit] fix import path

* [tests] lazy init test torchrec models

* [tests] update torch version in CI

* [tests] revert torch version in CI

* [tests] skip lazy init test
2023-03-17 13:49:04 +08:00
..
diffusers [tests] diffuser models in model zoo (#3136) 2023-03-14 17:20:28 +08:00
timm [test] added timm models to test model zoo (#3129) 2023-03-14 14:29:18 +08:00
torchaudio [tests] model zoo add torchaudio models (#3138) 2023-03-15 11:51:16 +08:00
torchrec [lazyinit] add correctness verification (#3147) 2023-03-17 13:49:04 +08:00
torchvision [test] added torchvision models to test model zoo (#3132) 2023-03-15 10:42:07 +08:00
transformers [booster] implemented mixed precision class (#3151) 2023-03-17 11:00:15 +08:00
__init__.py [test] add torchrec models to test model zoo (#3139) 2023-03-15 05:46:04 +00:00
registry.py [test] added torchvision models to test model zoo (#3132) 2023-03-15 10:42:07 +08:00