Toybrick

onnx 模型轉換 rknn 維度計算錯誤問題

arthurwu

中级会员

积分
201
发表于 2022-4-6 13:38:58    查看: 3646|回复: 2 | [复制链接]    打印 | 显示全部楼层
本帖最后由 arthurwu 于 2022-4-15 11:41 编辑

大家好

我有訓練一個 pytorch 模型並且想經由RK1808運算棒做計算
目前做法是先將 pytorch 模型轉成 onnx 模型 (opset11) ,再將 onnx 模型轉至 rknn
onnx 模型經由測試資料驗證,轉換與模型輸出結果皆正常而將onnx 轉成 rknn 時,輸出維度錯誤,不符合原本輸出

根據轉換後的結果(如圖所示)
permute輸出(500x150x1x1) 結果符合原始模型,並與 variable (220x500) 做矩陣相乘(MATRIXMUL)
正常結果應為 220 x 150 x 1 x 1
但是轉換後卻輸出 500 x 500 x 1 x 1


請問這部分該如何處理 ?

rknn toolkit 跟 npu driver 皆為 1.7.0

謝謝





本帖子中包含更多资源

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

x
回复

使用道具 举报

bobby_jiang

高级会员

积分
544
发表于 2022-4-26 09:49:38 | 显示全部楼层
先更新1.7.1试下,如果还不行,请提交完整的代码和模型。
回复

使用道具 举报

arthurwu

中级会员

积分
201
 楼主| 发表于 2022-4-27 10:30:35 | 显示全部楼层
bobby_jiang 发表于 2022-4-26 09:49
先更新1.7.1试下,如果还不行,请提交完整的代码和模型。

您好
我有嘗試過使用 1.7.1 結果也是相同
但是如果我把原本的 4 dimension (500 x 150 x 1 x 1)  改成  3 dimension (500 x 150 x 1) ,
這樣結果就會正常,
請問這是甚麼原因呢?
謝謝
回复

使用道具 举报

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

本版积分规则

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


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