code: def auto_wrap_big_layers(module: nn.Module, fsdp_config: AttrDict):
      """
      Automatically wrap the bigger layer in the module
      """
      with enable_wrap(
          auto_wrap_policy=_BigConvAutoWrapPolicy(
              fsdp_config.AUTO_WRAP_THRESHOLD
          ),
          wrapper_cls=_FSDP_WRAPPER,
          **fsdp_config,
      ):
          return auto_wrap(module)
  
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 7 for calling a global function with example usage: auto_wrap_big_layers(module, fsdp_config) and returns: auto_wrap