使用MO的時候 甚麼情況需要提供額外參數?
2022/08/05 下午 05:00
【第9堂】使用模型優化器轉換PyTorch/ONNX模型
黃仲民
觀看數：15
回答數：1
收藏數：0
請問一下,在使用MO轉換ONNX檔至IR時, 除了input_model必填,其他諸如--input,--mean_values這些是甚麼情況下需要給出? 不給對模型是否會造成影響
回答列表
-
2022/08/06 下午 02:46Chungyeh Wang贊同數：1不贊同數：0留言數：0
mo --input_model以外的參數取決於 被轉換的模型在使用原始深度學習框架建立模型與其訓練的程式碼 在處理模型輸入與輸出資料時做了甚麼. 這邊是一個例子, --mean_values or --scale_value 就是 原始深度學習框架 在處理模型輸入時 將輸入資料做normalization的程式碼所用的的數值. 如果不給這些值, 只接把資料輸入到模型去計算 模型的輸出就不是對的. 這像是計算機概論說的 garbege in, garbege out. 不是模型的問題, 是輸入的資料就不對了.