The simplest approach might be to load the state_dict into the model before adding the new layers. Alternatively, you could also add new key-value pairs into the state_dict as it’s derived from a dict.
1 Like