Toybrick

请教pytorch转rknn模型时遇到ConvTranspose的bug

changchenliang

注册会员

积分
125
楼主
发表于 2021-4-13 14:50:27    查看: 10305|回复: 1 | [复制链接]    打印 | 只看该作者
本帖最后由 changchenliang 于 2021-4-15 18:18 编辑

在pytorch(1.6.0)转rknn(1.6.0)模型时发现个疑似bug的问题:
在转化反卷积nn.ConvTranspose2d(in-channel, out-channel)时,貌似对输出out-channel这个参数无效。举例: nn.ConvTranspose2d(1024, 512, kernel_size=4, stride=2, padding=1, bias=True)时,输入了1024个channel的feature map,输出应该是512个channel的feature map, 但是发现转化后的rknn模型输出仍然是1024, 也就是说out-channel这个参数无效了。  因此就是报错(或者推理结果不对)    因此貌似在转化反卷积convtranspose2d时有bug 。  这种情况在以前pytorch1.2.0+rknn1.4.0版本时是没有的。。。。

因此能否请相关人员检查一下? 是否是bug啊?  否则convtranspose就没办法使用了

谢谢!




回复

使用道具 举报

changchenliang

注册会员

积分
125
沙发
 楼主| 发表于 2021-4-15 18:19:22 | 只看该作者
顶一下哈!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表