"...git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "15055f9a36d215e5a4fba8658e1b04c895881da6"
Commit e4bcff9b authored by lucasliunju's avatar lucasliunju Committed by binmakeswell
Browse files

[NFC] polish colossalai/builder/builder.py code style (#662)

parent 331683bf
...@@ -7,6 +7,7 @@ from collections.abc import Iterable ...@@ -7,6 +7,7 @@ from collections.abc import Iterable
from colossalai.registry import * from colossalai.registry import *
def build_from_config(module, config: dict): def build_from_config(module, config: dict):
"""Returns an object of :class:`module` constructed from `config`. """Returns an object of :class:`module` constructed from `config`.
...@@ -64,6 +65,7 @@ def build_from_registry(config, registry: Registry): ...@@ -64,6 +65,7 @@ def build_from_registry(config, registry: Registry):
return obj return obj
def build_layer(config): def build_layer(config):
"""Returns a layer object of :class:`nn.Module` constructed from `config`. """Returns a layer object of :class:`nn.Module` constructed from `config`.
...@@ -243,7 +245,6 @@ def build_lr_scheduler(config, optimizer): ...@@ -243,7 +245,6 @@ def build_lr_scheduler(config, optimizer):
config_['optimizer'] = optimizer config_['optimizer'] = optimizer
return build_from_registry(config_, LR_SCHEDULERS) return build_from_registry(config_, LR_SCHEDULERS)
def build_schedule(config): def build_schedule(config):
"""Returns a schedule of :class:`colossalai.engine.schedule.BaseSchedule`. """Returns a schedule of :class:`colossalai.engine.schedule.BaseSchedule`.
......
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