github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/kit/model_zoo/__init__.py

5 lines
136 B
Python
Raw Normal View History

[test] add torchrec models to test model zoo (#3139)
2023-03-15 05:46:04 +00:00
from . import diffusers, timm, torchaudio, torchrec, torchvision, transformers
[test] added timm models to test model zoo (#3129) * [test] added timm models to test model zoo * polish code * polish code * polish code * polish code * polish code
2023-03-14 06:29:18 +00:00
from .registry import model_zoo
__all__ = ['model_zoo']