class_docstr: This class implements `FastViT architecture <https://arxiv.org/pdf/2303.14189.pdf>`_
class_docstr_tok: ["This","class","implements","`","FastViT","architecture","<","https",":","\/\/arxiv.org\/pdf\/2303.14189.pdf",">","`","_"]
code: def _scrub_checkpoint(checkpoint, model):
      sterile_dict = {}
      for k1, v1 in checkpoint.items():
          if k1 not in model.state_dict():
              continue
          if v1.shape == model.state_dict()[k1].shape:
              sterile_dict[k1] = v1
      return sterile_dict
  
imports: import os
  import copy
  from functools import partial
  from typing import List, Tuple, Optional, Union
  import torch
  import torch.nn
  from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
  from timm.models.layers import DropPath, trunc_normal_
  from timm.models.registry import register_model
  from models.modules.mobileone import MobileOneBlock
  from models.modules.replknet import ReparamLargeKernelConv
label_desc: Use image node_id 3 for calling the FastViT obj's underlying member method code with example usage: obj._scrub_checkpoint(checkpoint, model) and returns: sterile_dict