想要保存训练的模型,使用到的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() 】