ColossalAI/colossalai/pipeline/__init__.py

4 lines
162 B
Python

from .pipelinable import PipelinableContext, PipelinableModel
from .layer_sepc import LayerSpec
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']