keras to onnx 問題

轉換了2個模型皆遇到問題:


因 model 是 tf.keras,因此先使用了工具:keras2onnx 轉換模型

https://github.com/onnx/keras-onnx


接著使用指令:

python /workspace/scripts/convert_model.py onnx in.onnx out.onnx


檢查模型:

km = ktc.ModelConfig(1001, "0001", "520", onnx_path=onnx_path)

eval_result = km.evaluate()

print("\nNpu performance evaluation result:\n" + str(eval_result))


因模型內有不支援的 node,兩個模型都刪除該 Transpose node:

m = onnx.load('out.onnx')

m = ktc.onnx_optimizer.delete_nodes(m, 'Transpose144_kn')



再次檢查模型,其中一個模型遇到問題:

Traceback 顯示在 km.evaluate 步驟出錯



另一個模型有得到輸出,但在下一步出錯:

ktc.kneron_inference([in_data], onnx_file=onnxName, input_names=["adjusted_input34_kn"])

得到 Segmentation fault (core dumped)


請問這兩個問題該如何排除?謝謝

Comments

  • Hi 楊堯順,


    方便提供下列項目:

    1. failed onnx model
    2. 可以重置toolchain問題的python script
    3. toolchain version

    以利我們定位錯誤原因嗎?


    thanks.

The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.