ColossalAI/colossalai/engine/paramhooks
Jiarui Fang 8d653af408 add a common util for hooks registered on parameter. (#292) 2022-03-11 15:50:28 +08:00
..
__init__.py add a common util for hooks registered on parameter. (#292) 2022-03-11 15:50:28 +08:00
_param_hookmgr.py add a common util for hooks registered on parameter. (#292) 2022-03-11 15:50:28 +08:00