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
fairscale
Commits
308f1057
"vscode:/vscode.git/clone" did not exist on "b64cbe323171ee02136995b543845a23cbf59524"
Unverified
Commit
308f1057
authored
Jun 23, 2021
by
Mehdi Mirzazadeh
Committed by
GitHub
Jun 23, 2021
Browse files
use ModuleLst for submodules (#722)
parent
1cc4c837
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
fairscale/experimental/nn/distributed_pipeline/graph.py
fairscale/experimental/nn/distributed_pipeline/graph.py
+1
-1
No files found.
fairscale/experimental/nn/distributed_pipeline/graph.py
View file @
308f1057
...
...
@@ -20,7 +20,7 @@ class MultiInputSequential(nn.Module):
def
__init__
(
self
,
*
modules
:
nn
.
Module
)
->
None
:
super
().
__init__
()
self
.
modules_list
=
modules
self
.
modules_list
=
nn
.
ModuleList
(
modules
)
def
forward
(
self
,
*
inputs
:
Tuple
[
Tensor
])
->
Tensor
:
# type: ignore
input
=
self
.
modules_list
[
0
](
*
inputs
)
...
...
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