Files
ColossalAI/colossalai/pipeline/__init__.py
Frank Lee 2b2dc1c86b [pipeline] refactor the pipeline module (#1087)
* [pipeline] refactor the pipeline module

* polish code
2022-06-10 11:27:38 +08:00

4 lines
162 B
Python

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