text: 
  import torch
  def load_state_dict_flexible(model, state_dict):
      try:
          model.load_state_dict(state_dict)
      except:
          print("Full loading failed!! Try partial loading!!")
  
      own_state = model.state_dict()
  
      for name, param in state_dict.items():
          if name not in own_state:
              print("Skipped: " + name)
              continue
          if isinstance(param, torch.nn.Parameter):
              # backwards compatibility for serialized parameters
              param = param.data
          try:
              own_state[name].copy_(param)
              print("Successfully loaded: "+name)
          except:
              print("Part load failed: " + name)
  
  def load_state_dict_flexible_with_fp16(model, state_dict):
      try:
          model.load_state_dict(state_dict)
      except:
          print("Full loading failed!! Try partial loading!!")
  
      own_state = model.state_dict()
  
      for name, param in state_dict.items():
          if name not in own_state:
              print("Skipped: " + name)
              continue
          if isinstance(param, torch.nn.Parameter):
              # backwards compatibility for serialized parameters
              param = param.data
          try:
              #print("Name {}, original_type: {}, load type".format(name, own_state[name].dtype, param.dtype))
              param = param.to(own_state[name].device)
              own_state[name].copy_(param)
              print("Successfully loaded: "+name)
          except:
              print("Part load failed: " + name)