Unverified Commit 203dd668 authored by Min Xu's avatar Min Xu Committed by GitHub
Browse files

[minor] add a warning in the doc (#1070)

parent eeb6684e
...@@ -879,6 +879,11 @@ class FullyShardedDataParallel(nn.Module): ...@@ -879,6 +879,11 @@ class FullyShardedDataParallel(nn.Module):
If you want the full param to be returned, you should call this function 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. 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) named_param = super().named_parameters(*args, **kwargs)
for name, param in named_param: for name, param in named_param:
......
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