ColossalAI/colossalai/gemini/ophooks/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .utils import BaseOpHook, register_ophooks_recursively

__all__ = ["BaseOpHook", "register_ophooks_recursively"]
[Gemini] clean no used MemTraceOp (#1970) 2022-11-17 05:41:54 +00:00			`from .utils import BaseOpHook, register_ophooks_recursively`
[zero] add sharded grad and refactor grad hooks for ShardedModel (#287) 2022-03-02 10:28:29 +00:00
[Gemini] clean no used MemTraceOp (#1970) 2022-11-17 05:41:54 +00:00			`__all__ = ["BaseOpHook", "register_ophooks_recursively"]`