Yes, but when I save optimizer state dict, I haven’t saved the parameters. So I need to save the parameters as well.
As for the model when I save the model’s state_dict I have saved all.
state = {
    'epoch': epoch,
    'state_dict': model.state_dict(),
    'optimizer': optimizer.state_dict(),
    #add params in here also...
}
torch.save(state, filepath)

Would be a decent save.