yes I tried

torch.save(model, ‘path_name.pth’)

and it seemed to work.
thanks @John_Deterious
and whats the best way to save the state of the optimizers ? why do we even need to save optimizer’s state? Can you pls explain this to me ?
Thanks in advance !!!