ColossalAI/colossalai/gemini/ophooks
Zihao 38ea4ba1bd
[Gemini] fix grad unreleased issue and param recovery issue (#2052)
2022-12-02 16:04:19 +08:00
..
__init__.py
_shard_grad_ophook.py [hotfix] remove potiential circle import (#1307) 2022-07-14 13:44:26 +08:00
_shard_param_ophook.py
mem_trace_hook.py [Gemini] add bert for MemtracerWrapper unintests (#1982) 2022-11-18 14:58:28 +08:00
param_trace_hook.py [Gemini] fix grad unreleased issue and param recovery issue (#2052) 2022-12-02 16:04:19 +08:00
utils.py