github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/engine/paramhooks/__init__.py

4 lines
77 B
Python
Raw Normal View History

add a common util for hooks registered on parameter. (#292)
2022-03-02 06:38:22 +00:00
from ._param_hookmgr import BaseParamHookMgr
Fix/format colossalai/engine/paramhooks/(#350)
2022-03-09 09:28:17 +00:00
__all__ = ["BaseParamHookMgr"]