問題タブ [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 問題
0
投票する
2
に答える
7985
参照
pytorch - pytorch にインポートされた onnx モデルを実行できませんか?
onnx 形式からモデルをインポートして pytorch を操作しようとしています。同じ例を得るのが難しいと感じています。インターネットのリソースのほとんどは、pytorch モデルを onnx にエクスポートすることについて語っています。
torch.onnx() はモデルをエクスポートすることしかできず、インポート メソッドはまだ実装されていないことがわかりました。onnx ライブラリを直接インストールすると、onnx.load("model_name.onnx") を実行するのに役立ちます。このモデルを pytorch で使用するにはどうすればよいですか? model.to(device="GPU") でモデルを GPU に移動できません