ColossalAI/colossalai/engine/paramhooks/_param_hookmgr.py at af2cfed44769dfd822edbdfff0052472c69f55fc - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-28 11:01:15 +00:00

Files

Jiarui Fang 6f22fb1906 add a common util for hooks registered on parameter. (#292 )

2022-03-02 14:38:22 +08:00

1.1 KiB

Raw Blame History

View Raw