I have a simple problem: I want to transfer the weights of one model to another, such that only the final layers of the model differ, an MLP for example.
I am loading the pretrained model with `load_state_dict, but it tries to copy ALL the weights of the pretrained model. How do I copy only the weights that I want, or that are existing in the model (aka, same layer name)?