ColossalAI/colossalai/nn/layer/wrapper/__init__.py

5 lines
160 B
Python

from .lambda_wrapper import LambdaWrapper
from .pipeline_wrapper import PipelineSharedModuleWrapper
__all__ = ['LambdaWrapper', 'PipelineSharedModuleWrapper']