倾_尘
2022-06-30 16:16
采纳率: 0%
浏览 12

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

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

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

我想要达到的结果

成功保存

  • 写回答
  • 好问题 提建议
  • 追加酬金
  • 关注问题
  • 邀请回答

1条回答 默认 最新

相关推荐 更多相似问题