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']