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
203dd668
Unverified
Commit
203dd668
authored
Sep 12, 2022
by
Min Xu
Committed by
GitHub
Sep 12, 2022
Browse files
[minor] add a warning in the doc (#1070)
parent
eeb6684e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
0 deletions
+5
-0
fairscale/nn/data_parallel/fully_sharded_data_parallel.py
fairscale/nn/data_parallel/fully_sharded_data_parallel.py
+5
-0
No files found.
fairscale/nn/data_parallel/fully_sharded_data_parallel.py
View file @
203dd668
...
...
@@ -879,6 +879,11 @@ class FullyShardedDataParallel(nn.Module):
If you want the full param to be returned, you should call this function
under a `summon_full_params` context when using flattened or original params.
.. warning:: This overloaded method will *not* be called in the case of a parent module
containing a FSDP-wrapped child module. Calling parent.named_parameters()
*will* return original *unclean* key strings, i.e. _fsdp_wrapped_module and
_fpw_module are included the key string.
"""
named_param
=
super
().
named_parameters
(
*
args
,
**
kwargs
)
for
name
,
param
in
named_param
:
...
...
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