Load model weights into a new created model

5

You could iterate the state_dict and change the name of these keys as shown here.