code: def convert_state_dict(state_dict: OrderedDict):
      new_state_dict = OrderedDict()
      for k, v in state_dict.items():
          k = k.replace("dwconv.dwconv", "dwconv")
          new_state_dict[k] = v
      return new_state_dict
  
imports: from functools import partial
  from torch import nn
  from collections import OrderedDict
label_desc: Use image node_id 2 for calling a global function with example usage: convert_state_dict(state_dict) and returns: new_state_dict