github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 12:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] add torchrec models to test model zoo (#3139)

Browse Source
This commit is contained in:
YuliangLiu0306
2023-03-15 13:46:04 +08:00
committed by GitHub
parent 14a115000b
commit ecd643f1e4
4 changed files with 200 additions and 160 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/kit/model_zoo/__init__.py
View File

@@ -1,4 +1,5 @@
from . import diffusers, timm, torchaudio, torchvision, transformers
from . import diffusers, timm, torchaudio, torchrec, torchvision, transformers
from .registry import model_zoo
__all__ = ['model_zoo']