2

onnx 形式からモデルをインポートして pytorch を操作しようとしています。同じ例を得るのが難しいと感じています。インターネットのリソースのほとんどは、pytorch モデルを onnx にエクスポートすることについて語っています。

torch.onnx() はモデルをエクスポートすることしかできず、インポート メソッドはまだ実装されていないことがわかりました。onnx ライブラリを直接インストールすると、onnx.load("model_name.onnx") を実行するのに役立ちます。このモデルを pytorch で使用するにはどうすればよいですか? model.to(device="GPU") でモデルを GPU に移動できません

4

2 に答える 2

4

onnx モデルを PyTorch に変換するのに役立つonnx2pytorchというライブラリがあります。

于 2021-01-20T17:33:07.670 に答える