I do not know how to apply the answer by @ptrblck to this case

  
    
    
    Missing keys & unexpected keys in state_dict when loading self trained model vision
  
  
    Could you try to load the state_dict using model.load_state_dict(torch.load(PATH), strict=False)?