[Pytorch] CurveNet執行pytorch2onnx.py輸出onnx格式時出錯
為了我們的作品，我們找了一個pytorch模型來使用，這是用來預測點雲的模型。預計先在colab訓練，得到一個.pth檔後，再將其轉成ONNX檔以利後續用在KL-520加速棒上。但在轉成ONNX時就遇到了滿多錯誤的，以下是主要的內容。如果有什麼描述的不清楚的請跟我說，謝謝。
The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.
Comments
Hello,
有關onnx生成的部分，推測你遇到的問題與這篇討論串相同 https://github.com/pytorch/pytorch/issues/34780#issuecomment-876969861
把curvenet_utils.py當中MaskedMaxPool作下列下列紅框處修改應可以使用它們提供的pytoch2onnx.py生出onnx
但要注意onnx生成時間很久, 代表graph很大, 懷疑當中有非常多不支援op, 我用netron甚至打不開, 後續應要做更多處理
關於這個問題:
請確認有兩個cuda要改成cpu:
你好，照你的方式修改，有成功轉出ONNX檔了，謝謝!