我们不能运行导入到pytorch的onnx模型吗？

Question

我一直在尝试从onnx格式导入模型以使用pytorch。我发现很难获得相同的例子。互联网上的大多数资源都在谈论将pytorch模型导出到onnx。

我发现torch.onnx（）只能导出模型，而导入方法尚未实现。直接安装onnx库，可以帮助我执行onnx.load（“ model_name.onnx”）。如何在pytorch中使用此模型？我无法通过model.to（device =“ GPU”）将模型移至GPU

Answer 1

有一个名为 onnx2pytorch 的库可以帮助您将 onnx 模型转换为 PyTorch。

Answer 2

PyTorch当前不支持导入onnx模型。在撰写此答案时，它是open feature request。

虽然不能保证正常工作。潜在的解决方案是使用Microsoft开发的名为MMdnn的工具（不是仅限Windows！），该工具支持与各种框架之间的转换。不幸的是，onnx只能是转换的目标，而不能是源。也就是说，您也许能够将模型导入到另一个框架，然后使用MMdnn从该框架转换为pytorch。显然，这不是理想的，成功的潜力将取决于其他框架如何使用onnx，而这可能不适合MMdnn的工作方式。