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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-10 15:55:36 +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