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)
請問這兩個問題該如何排除?謝謝
The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.
Comments
Hi 楊堯順,
方便提供下列項目:
以利我們定位錯誤原因嗎?
thanks.