模型保存问题

[WenwenZong]

在代码中提供了保存模型参数的代码，请问各位复现代码的大佬，在实现断点续训时，如何保存optimizer的参数？在采用保存语句时，会出现以下错误。
"optimizer_state": optim.state_dict(),
AttributeError: 'Optimizer' object has no attribute 'state_dict'
请各位解决下疑问，万分感谢。