Have a look at the ImageNet example 6 to see, how to store all necessary information in order to be able to continue the training.
Also, I would recommend to save and load the state_dict instead of the complete model.