pytorch模型轉onnx,報錯kind_.is_prim() INTERNAL ASSERT FAILED at /pytorch/torch/csrc/jit/ir.cpp:922

在將pytorch模型轉爲onnx模型時遇到一個報錯如下: 這個報錯指向範圍很寬泛,代碼有幾種不同的錯誤都會報這個錯誤,我在本文裏面列出我遇到的這個報錯但是實際上是不同類型的代碼問題以及解決方法,目前我只遇到兩個報錯相同但實際上代碼錯誤不同的問題,未來遇到還會補充。 1.創建tensor的問題   我遇到的第一個這個報錯是在使用torch.Tensor()方法創建新的tensor,最終的解決方法是
相關文章
相關標籤/搜索