I’m using to save my model and re-load it these functions,
model.load_state_dict(torch.load())  
torch.save(model.state_dict())

I want to add a new layer to my model and re-save it again as a new model instead of calling two models in my new model. how can I load the parameters only to the pre-trained part in the new model that matches the old model ?