@smth’s answer is the solution to this.
Use - model.load_state_dict(torch.load(PATH)) and for more details refer to the documentation (https://pytorch.org/tutorials/beginner/saving_loading_models.html 177)