albanD:

o save the opti


What about scheduler.state_dict().
Which parameters are the important in order to restart model from the checkpoint?
(model.state_dict(), optimizer.state_dict(), scheduler.state_dict())   all these are important to restart model from checkpoint?