code: def FSDPLinearEvalMLP(
      model_config: AttrDict,
      in_channels: int,
      dims: List[int],
      use_bn: bool = False,
      use_relu: bool = False,
  ):
      mlp = LinearEvalMLP(
          model_config, in_channels, dims, use_bn, use_relu
      )
      mlp = fsdp_auto_wrap_bn(mlp)
      return fsdp_wrapper(mlp, **model_config.FSDP_CONFIG)
  
file_path: vissl/vissl/models/heads/linear_eval_mlp.py
imports: from typing import List
  import torch
  import torch.nn
  from vissl.config import AttrDict
  from vissl.models.heads import register_model_head
  from vissl.models.heads.mlp import MLP
  from vissl.models.heads.mlp_res import MLPResidual
  from vissl.utils.fsdp_utils import fsdp_auto_wrap_bn, fsdp_wrapper
label_desc: Use image node_id 1 for calling a global function with example usage: FSDPLinearEvalMLP(model_config, in_channels, dims, use_bn, use_relu) and returns: fsdp_wrapper