As the title, I have used torch.save(model.state_dict()) when try to save the model. I checked that only weights and biases should be saved. At that time, other parameters such as optimizer were not saved during training. How to continue training? Try to use model.load_state_dict(torch.load(***.pkl), but I don’t know then. It’s best to use the relevant code for reference.