Toybrick

标题: rknntoolkit 1.6.1转化pytorch的奇怪bug [打印本页]

作者: jiexu16 时间: 2021-7-21 08:53
标题: rknntoolkit 1.6.1转化pytorch的奇怪bug
在使用rknntoolkit 1.6.1转化pytorch（Torch 1.6.0）到RKNN时出现了一个奇怪的BUG，在矩阵乘前面的Transpose在转化过程中一会儿有，一会儿没有，对比图如下，我用pytorch转化为onnx来看的模型：

这个是转化后Transpose被删除：

[attach]1936[/attach]

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

[attach]1935[/attach]

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

[attach]1937[/attach]

作者: jiexu16 时间: 2021-7-21 09:41
请问有可以手动修改RKNN模型的工具吗

欢迎光临 Toybrick (https://t.rock-chips.com/)