ColossalAI/colossalai/builder/__init__.py

6 lines
172 B
Python

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