Toybrick

标题: onnx 模型轉換 rknn 維度計算錯誤問題 [打印本页]

作者: arthurwu 时间: 2022-4-6 13:38
标题: onnx 模型轉換 rknn 維度計算錯誤問題
本帖最后由 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

謝謝

作者: bobby_jiang 时间: 2022-4-26 09:49
先更新1.7.1试下，如果还不行，请提交完整的代码和模型。

作者: arthurwu 时间: 2022-4-27 10:30

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) ，
這樣結果就會正常，
請問這是甚麼原因呢?
謝謝

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