Ok thank you very much for your reply,I’m doing it.
Also one more thing, I do need to save the accuracy and epoch along the model.
Will it be saved by torch.save(the_model.state_dict(), PATH) itself. If so then how to access that.