ColossalAI/colossalai/zero/utils/zero_hook.py at 36ab2cb783fdee9899f734f99633fc0d63d4e980 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 18:14:16 +00:00

Files

Jiarui Fang 2938edf446 [Gemini] update the non model data record method in runtime memory tracer (#2128 )

2022-12-13 17:11:31 +08:00

4.7 KiB

Raw Blame History

View Raw