mirror of https://github.com/hpcaitech/ColossalAI
parent
314d8c497f
commit
63f250bbd4
|
@ -1,4 +1,4 @@
|
|||
from .pipelinable import PipelinableContext, PipelinableModel
|
||||
from .layer_sepc import LayerSpec
|
||||
from .layer_spec import LayerSpec
|
||||
|
||||
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']
|
|
@ -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):
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
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
|
||||
|
|
Loading…
Reference in New Issue