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

Question

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

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

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

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

运行结果及报错内容

TypeError : can't pickle mappingproxy objects

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

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

我想要达到的结果

成功保存

Answer 1

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