ColossalAI/tests/kit/model_zoo/torchrec
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
..
__init__.py [lazyinit] add correctness verification (#3147) 2023-03-17 13:49:04 +08:00
torchrec.py [test] fixed torchrec registration in model zoo (#3177) 2023-03-20 16:19:06 +08:00