code: def fsdp_recursive_reset_lazy_init(fsdp_module: FSDP):
      """
      Before the first forward pass, an FSDP module might have been initialized
      for instance by calling load_state_dict or load_local_state_dict to
      reload a previous training checkpoint.
  
      This function will recursively walk though the sub-FSDP modules and
      call _reset_lazy_init on each of them.
      """
      for module in fsdp_module.modules():
          if isinstance(module, FSDP) and module._is_root is not None:
              module._reset_lazy_init()
  
file_path: vissl/vissl/utils/fsdp_utils.py
imports: import torch.distributed
  import torch.nn
  from fairscale.nn import auto_wrap, default_auto_wrap_policy, enable_wrap
  from fairscale.nn.data_parallel import auto_wrap_bn, FullyShardedDataParallel
  from vissl.config.attr_dict import AttrDict
label_desc: Use image node_id 1 for calling a global function with example usage: fsdp_recursive_reset_lazy_init(fsdp_module) without return types