github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '258b43317c'
${ noResults }
ColossalAI/tests/kit/model_zoo
History
Frank Lee 085e7f4eff
[test] fixed torchrec registration in model zoo (#3177) 		2 years ago
..
diffusers [tests] diffuser models in model zoo (#3136) 2 years ago
timm [test] added timm models to test model zoo (#3129) 2 years ago
torchaudio [tests] model zoo add torchaudio models (#3138) 2 years ago
torchrec [test] fixed torchrec registration in model zoo (#3177) 2 years ago
torchvision [test] added torchvision models to test model zoo (#3132) 2 years ago
transformers [booster] implemented mixed precision class (#3151) 2 years ago
__init__.py [test] add torchrec models to test model zoo (#3139) 2 years ago
registry.py [test] added torchvision models to test model zoo (#3132) 2 years ago