pytorch如何保存模型训练参数,torch.save报错

问题遇到的现象和发生背景

想要保存训练的模型,使用到的torch.save()报错

问题相关代码,请勿粘贴截图

torch.save(self.model,'{}{}_{}.pth ' .format(self.save_path,self.model_name,self.log_name))

运行结果及报错内容

TypeError : can't pickle mappingproxy objects

我的解答思路和尝试过的方法

把这句注释掉可以运行但是不能保存模型

我想要达到的结果

成功保存

看下这篇博客,也许你就懂了,链接:PyTorch保存网络结构以及参数【 torch.save()、torch.load() 】