github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-19 12:12:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix file name (#1759)

Browse Source 
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/pipeline/__init__.py
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']

0
colossalai/pipeline/layer_sepc.py → colossalai/pipeline/layer_spec.py
View File

2
colossalai/pipeline/pipelinable.py
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):

2
docs/colossalai/colossalai.pipeline.layer_sepc.rst
View File

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

2
docs/colossalai/colossalai.pipeline.rst
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