logo
Loading...

使用MO的時候 甚麼情況需要提供額外參數? - Cupoy

請問一下,在使用MO轉換ONNX檔至IR時,除了input_model必填,其他諸如--input,...

使用MO的時候 甚麼情況需要提供額外參數?

2022/08/05 下午 05:00
【第9堂】使用模型優化器轉換PyTorch/ONNX模型
黃仲民
觀看數:13
回答數:1
收藏數:0

請問一下,在使用MO轉換ONNX檔至IR時, 除了input_model必填,其他諸如--input,--mean_values這些是甚麼情況下需要給出? 不給對模型是否會造成影響

回答列表

  • 2022/08/06 下午 02:46
    Chungyeh Wang
    贊同數:1
    不贊同數:0
    留言數:0

    mo --input_model以外的參數取決於 被轉換的模型在使用原始深度學習框架建立模型與其訓練的程式碼 在處理模型輸入與輸出資料時做了甚麼. 這邊是一個例子, --mean_values or --scale_value 就是 原始深度學習框架 在處理模型輸入時 將輸入資料做normalization的程式碼所用的的數值. 如果不給這些值, 只接把資料輸入到模型去計算 模型的輸出就不是對的. 這像是計算機概論說的 garbege in, garbege out. 不是模型的問題, 是輸入的資料就不對了.