I used this
for k,v in model_dict.items():
  if(k in trained_original_state_dict):
    model_dict[k] = trained_original_state_dict[k]

to copy the weights from one model to another
i.e from pretrained model to the new model