code: def load_state_dict(self, state_dict, strict=True):
      # '.'s are no longer allowed in module names, but previous _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.
      # Credit - https://github.com/pytorch/vision/blob/master/torchvision/models/densenet.py#def _load_state_dict()
      import re
  
      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 super(ImageClassifier, self).load_state_dict(
          state_dict, strict
      )
  
imports: from torchvision.models.densenet import DenseNet
label_desc: Use image node_id 2 for calling the ImageClassifier obj's underlying member method code with example usage: obj.load_state_dict(state_dict, strict) and returns: super