github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-08 03:24:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
feature/stable-diffusion
ColossalAI/tests/kit/model_zoo/torchrec
History
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