ColossalAI/tests/kit/model_zoo
Frank Lee 085e7f4eff
[test] fixed torchrec registration in model zoo (#3177)
* [test] fixed torchrec registration in model zoo

* polish code

* polish code

* polish code
2023-03-20 16:19:06 +08:00
..
diffusers
timm
torchaudio
torchrec [test] fixed torchrec registration in model zoo (#3177) 2023-03-20 16:19:06 +08:00
torchvision
transformers [booster] implemented mixed precision class (#3151) 2023-03-17 11:00:15 +08:00
__init__.py
registry.py