Unverified Commit 63f250bb authored by Ziyue Jiang's avatar Ziyue Jiang Committed by GitHub
Browse files

fix file name (#1759)


Co-authored-by: default avatarZiyue Jiang <ziyue.jiang@gmail.com>
parent 314d8c49
from .pipelinable import PipelinableContext, PipelinableModel
from .layer_sepc import LayerSpec
from .layer_spec import LayerSpec
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']
\ No newline at end of file
......@@ -9,7 +9,7 @@ from colossalai.nn.layer.utils import CheckpointModule
from colossalai.tensor import ColoParameter
from colossalai.core import global_context as gpc
from colossalai.context import ParallelMode
from .layer_sepc import LayerSpec
from .layer_spec import LayerSpec
class PipelinableContext(InsertPostInitMethodToModuleSubClasses):
......
colossalai.pipeline.layer\_sepc
===============================
.. automodule:: colossalai.pipeline.layer_sepc
.. automodule:: colossalai.pipeline.layer_spec
:members:
......@@ -8,6 +8,6 @@ colossalai.pipeline
.. toctree::
:maxdepth: 2
colossalai.pipeline.layer_sepc
colossalai.pipeline.layer_spec
colossalai.pipeline.pipelinable
colossalai.pipeline.utils
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment