torch.save时报错 Can‘t pickle local object ‘_initialize.<locals>.ResBasicBlock.__init__.<locals>.lambda

今天在保存模型时遇到了如下所示的问题,查找了很多资料,没有满意的结果。

torch.save时报错 Can‘t pickle local object ‘_initialize.<locals>.ResBasicBlock.__init__.<locals>.lambda

后面发现是pickle有限制,无法解析我模型文件中的lambda表达式

进入模型文件修改无法pickle的位置即可:

如图,我的是

ResBasicBlock.__init__下的lambda表达式,后面换成了函数

torch.save时报错 Can‘t pickle local object ‘_initialize.<locals>.ResBasicBlock.__init__.<locals>.lambda_第1张图片

你可能感兴趣的:(pytorch)