ColossalAI/colossalai/legacy/builder/__init__.py

4 lines
166 B
Python

from .builder import build_from_config, build_from_registry, build_gradient_handler
__all__ = ["build_gradient_handler", "build_from_config", "build_from_registry"]