AttributeError: 'Conv2d' object has no attribute 'padding_mode'问题解决

在一个预训练模型加载过程中,发现碰到这样的一个错误:AttributeError: 'Conv2d' object has no attribute 'padding_mode'。

找了相关资料,发现其问题为pytorch版本的问题,即原来的预训练模型为pytorch 1.0版本环境下训练编译的,而现在的开发环境为pytorch 1.1版本。

网上很多办法都是直接退回到pytorch1.0版本。

这种比较笨的办法,为了实现更加简单的操作,调试到conv2d这种函数的forward中,主要是其中的self.paddingmode未赋值,导致的退出。

手动添加一句,发现即可实现运行,如下:

AttributeError: 'Conv2d' object has no attribute 'padding_mode'问题解决_第1张图片

不过上述过程需要在调试完成之后,进行去掉,否则可能会影响其他程序。当然也可以将程序模型再重新保存一份即可。

你可能感兴趣的:(pytorch)