diff --git a/colossalai/pipeline/__init__.py b/colossalai/pipeline/__init__.py index 625bd7ef5..0fcde9707 100644 --- a/colossalai/pipeline/__init__.py +++ b/colossalai/pipeline/__init__.py @@ -1,4 +1,4 @@ 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 diff --git a/colossalai/pipeline/layer_sepc.py b/colossalai/pipeline/layer_spec.py similarity index 100% rename from colossalai/pipeline/layer_sepc.py rename to colossalai/pipeline/layer_spec.py diff --git a/colossalai/pipeline/pipelinable.py b/colossalai/pipeline/pipelinable.py index 4d37c9833..9731530a6 100644 --- a/colossalai/pipeline/pipelinable.py +++ b/colossalai/pipeline/pipelinable.py @@ -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): diff --git a/docs/colossalai/colossalai.pipeline.layer_sepc.rst b/docs/colossalai/colossalai.pipeline.layer_sepc.rst index 0ff6a83c2..156660b5c 100644 --- a/docs/colossalai/colossalai.pipeline.layer_sepc.rst +++ b/docs/colossalai/colossalai.pipeline.layer_sepc.rst @@ -1,5 +1,5 @@ colossalai.pipeline.layer\_sepc =============================== -.. automodule:: colossalai.pipeline.layer_sepc +.. automodule:: colossalai.pipeline.layer_spec :members: diff --git a/docs/colossalai/colossalai.pipeline.rst b/docs/colossalai/colossalai.pipeline.rst index adaebea2d..6f7652d49 100644 --- a/docs/colossalai/colossalai.pipeline.rst +++ b/docs/colossalai/colossalai.pipeline.rst @@ -8,6 +8,6 @@ colossalai.pipeline .. toctree:: :maxdepth: 2 - colossalai.pipeline.layer_sepc + colossalai.pipeline.layer_spec colossalai.pipeline.pipelinable colossalai.pipeline.utils