Toybrick

rknntoolkit 1.6.1转化pytorch的奇怪bug

jiexu16

注册会员

积分
64
发表于 2021-7-21 08:53:40    查看: 3836|回复: 1 | [复制链接]    打印 | 显示全部楼层
在使用rknntoolkit 1.6.1转化pytorch(Torch 1.6.0)到RKNN时出现了一个奇怪的BUG,在矩阵乘前面的Transpose在转化过程中一会儿有,一会儿没有,对比图如下,我用pytorch转化为onnx来看的模型:

这个是转化后Transpose被删除:




这个是转化后Transpose多了一个:




模型转化为RKNN运行时报错如下:对应着多了一个Transpose的报错,可以从报错LOG中计算得到如果没有多个那个Transpose,转化的SIZE是相等的





本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jiexu16

注册会员

积分
64
 楼主| 发表于 2021-7-21 09:41:13 | 显示全部楼层
请问有可以手动修改RKNN模型的工具吗
回复

使用道具 举报

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

本版积分规则

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


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