Parameters:
         * module (torch.nn.Module) -- Root module.
     * **state_dict_type** (*StateDictType*) -- the desired
       "state_dict_type" to set.

     * **state_dict_config** (*Optional**[**StateDictConfig**]*)
       -- the configuration for the target "state_dict_type".

  Return type:
     *Generator*

static summon_full_params(module, recurse=True, writeback=True, rank0_only=False, offload_to_cpu=False, with_grads=False)
  A context manager to expose full params for FSDP instances. Can
  be useful *after* forward/backward for a model to get the params
  for additional processing or checking. It can take a non-FSDP
  module and will summon full params for all contained FSDP
  modules as well as their children, depending on the "recurse"
  argument.

  Note:

    This can be used on inner FSDPs.

  Note:

    This can *not* be used within a forward or backward pass. Nor
