AS state_dict states, it saves a dictionary. When you load an state dict, pytorch matchs layers by name. If you change names you won’t be able to automatically load previous state dicts.
To do so, you have to load the dict and then iterate over keys changing their name to match new model names.