fix file name (#1759)

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
pull/1760/head
Ziyue Jiang 2022-10-25 16:48:48 +08:00 committed by GitHub
parent 314d8c497f
commit 63f250bbd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 4 additions and 4 deletions

View File

@ -1,4 +1,4 @@
from .pipelinable import PipelinableContext, PipelinableModel
from .layer_sepc import LayerSpec
from .layer_spec import LayerSpec
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']

View 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):

View File

@ -1,5 +1,5 @@
colossalai.pipeline.layer\_sepc
===============================
.. automodule:: colossalai.pipeline.layer_sepc
.. automodule:: colossalai.pipeline.layer_spec
:members:

View File

@ -8,6 +8,6 @@ colossalai.pipeline
.. toctree::
:maxdepth: 2
colossalai.pipeline.layer_sepc
colossalai.pipeline.layer_spec
colossalai.pipeline.pipelinable
colossalai.pipeline.utils