code: def update_state_dict(state_dict):
      # '.'s are no longer allowed in module names, but pervious _DenseLayer
      # has keys 'norm.1', 'relu.1', 'conv.1', 'norm.2', 'relu.2', 'conv.2'.
      # They are also in the checkpoints in model_urls. This pattern is used
      # to find such keys.
      pattern = re.compile(
          r"^(.*denselayer\d+\.(?:norm|relu|conv))\.((?:[12])\.(?:weight|bias|running_mean|running_var))$"
      )
      for key in list(state_dict.keys()):
          res = pattern.match(key)
          if res:
              new_key = res.group(1) + res.group(2)
              state_dict[new_key] = state_dict[key]
              del state_dict[key]
      return state_dict
  
imports: from __future__ import print_function, division, absolute_import
  import torchvision.models
  import torch.utils.model_zoo
  import torch.nn.functional
  import types
  import re
label_desc: Use image node_id 1 for calling a global function with example usage: update_state_dict(state_dict) and returns: state_dict