ColossalAI/colossalai/pipeline/__init__.py

4 lines
162 B
Python
Raw Normal View History

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