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 .pipelinable import PipelinableContext, PipelinableModel
from .layer_sepc import LayerSpec from .layer_spec import LayerSpec
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec'] __all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']
\ No newline at end of file
...@@ -9,7 +9,7 @@ from colossalai.nn.layer.utils import CheckpointModule ...@@ -9,7 +9,7 @@ from colossalai.nn.layer.utils import CheckpointModule
from colossalai.tensor import ColoParameter from colossalai.tensor import ColoParameter
from colossalai.core import global_context as gpc from colossalai.core import global_context as gpc
from colossalai.context import ParallelMode from colossalai.context import ParallelMode
from .layer_sepc import LayerSpec from .layer_spec import LayerSpec
class PipelinableContext(InsertPostInitMethodToModuleSubClasses): class PipelinableContext(InsertPostInitMethodToModuleSubClasses):
......
colossalai.pipeline.layer\_sepc colossalai.pipeline.layer\_sepc
=============================== ===============================
.. automodule:: colossalai.pipeline.layer_sepc .. automodule:: colossalai.pipeline.layer_spec
:members: :members:
...@@ -8,6 +8,6 @@ colossalai.pipeline ...@@ -8,6 +8,6 @@ colossalai.pipeline
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.pipeline.layer_sepc colossalai.pipeline.layer_spec
colossalai.pipeline.pipelinable colossalai.pipeline.pipelinable
colossalai.pipeline.utils 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