rknntoolkit 1.6.1转化pytorch的奇怪bug

[复制链接] · 发表于 2021-7-21 08:53:40

在使用rknntoolkit 1.6.1转化pytorch（Torch 1.6.0）到RKNN时出现了一个奇怪的BUG，在矩阵乘前面的Transpose在转化过程中一会儿有，一会儿没有，对比图如下，我用pytorch转化为onnx来看的模型：

这个是转化后Transpose被删除：

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

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

只看该作者 · 发表于 2021-7-21 09:41:13

请问有可以手动修改RKNN模型的工具吗

rknntoolkit 1.6.1转化pytorch的奇怪bug

本帖子中包含更多资源