I trained my model on two classes. I saved the model weights and I want to initialise my new model with the saved weights . How to do it ?
If you’ve saved the state_dict you can load it with model.load_state_dict(torch.load(PATH_TO_DICT)).
Have a look at the Serialization Tutorial for more information.