Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
e4bcff9b
"...git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "15055f9a36d215e5a4fba8658e1b04c895881da6"
Commit
e4bcff9b
authored
Apr 03, 2022
by
lucasliunju
Committed by
binmakeswell
Apr 06, 2022
Browse files
[NFC] polish colossalai/builder/builder.py code style (#662)
parent
331683bf
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
colossalai/builder/builder.py
colossalai/builder/builder.py
+2
-1
No files found.
colossalai/builder/builder.py
View file @
e4bcff9b
...
@@ -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`.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment