logo
Loading...

使用模型優化器轉換PyTorch/ONNX模型 - Cupoy

在執行torch.onnx.export時出現RuntimeError: ONNX export f...

使用模型優化器轉換PyTorch/ONNX模型

2022/07/30 下午 01:01
Edge AI 工程師的20堂必修課
jj
觀看數：79
回答數：2
收藏數：0

在執行torch.onnx.export時出現 RuntimeError: ONNX export failed: Couldn't export Python operator HardSwishJitAutoFn PYTHO==3.6 torch==1.7 ![螢幕擷取畫面 2022-07-30 130110.png](http://kwassistfile.cupoy.com/000001824D79ABDE000000046375706F795F72656C656173655155455354/1657177895738/large)

回答列表

  • 2022/07/31 下午 09:21
    Chungyeh Wang
    贊同數：0
    不贊同數：0
    留言數：0

    不是所有的pytorch function都可以轉成ONNX. 請確認HardSwishJitAutoFn是否可以轉成ONNX. 一般來說,一些新的operator需要新版的pytorch and onnx python packages.

  • 2022/08/01 下午 04:25
    jj
    贊同數：0
    不贊同數：0
    留言數：1

    去哪裡確認HardSwishJitAutoFn是否可以轉成ONNX 我是使用102-pytorch-onnx-to-openvino.ipynb 出現這樣的問題 且我的torch是1.10.2 我在googlecolab也遇到依樣問題 https://drive.google.com/file/d/1KXQ4htTNgJOZUjq2gmHWTerE5OXOTTSq/view?usp=sharing