How resume the saved trained model at specific epoch

oh bummer! Hope it won’t take too long to rerun them. Also, you can consider saving the state of optimizer. A good example is here: Saving and loading a model in Pytorch?

3 Likes