ColossalAI/tests/kit
Frank Lee 1ad3a636b1
[test] fixed torchrec model test (#3167)
* [test] fixed torchrec model test

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-03-20 11:40:25 +08:00
..
model_zoo [test] fixed torchrec model test (#3167) 2023-03-20 11:40:25 +08:00
__init__.py [test] added timm models to test model zoo (#3129) 2023-03-14 14:29:18 +08:00