問題タブ [onnx]

質問する

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

505 問題
Newest
Active
Bountied
318
Unanswered
0 投票する
1 に答える
3078 参照

pytorch - ONNX と pytorch で出力が異なる

私は pytorch Resnet50 モデルを ONNX に変換して推論しようとしています。変換手順ではエラーは発生しませんが、onnxruntime からの onnx モデルの最終結果は、pytorch からの元のモデルの結果と大きなギャップがあります。
可能な解決策は何ですか?

ONNX のバージョン: 1.5.0 pytorch
のバージョン: 1.1.0
CUDA: 9.0
システム: Ubuntu 18.06
Python: 3.5

変換コードはこちら

これが結果チェックコードです

そして、チェックコードの出力（いくつかの警告がありますが、問題ではないと思います）は


12345678910