Hi,
Usually you save all the info of your model given by model.state_dict(). This contains the Parameters and Buffers.
If your optimizer also has a state that is important to maintain, you can also save the optimizer’s state_dict().