前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >AttributeError: 'Conv2d' object has no attribute 'padding_mode'问题解决

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

作者头像
sparkexpert
发布2019-09-18 11:05:58
2.7K0
发布2019-09-18 11:05:58
举报

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

本文链接:https://blog.csdn.net/sparkexpert/article/details/100296305

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

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

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

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

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

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

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年09月02日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档