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

Question

我一直在尝试从 onnx 格式导入模型以使用 pytorch。我发现很难找到同样的例子。由于 Internet 上的大多数资源都在谈论将 pytorch 模型导出到 onnx。

发现torch.onnx() 只能导出模型，导入方法还没有实现。直接安装 onnx 库，帮助我做 onnx.load("model_name.onnx")。如何将此模型与 pytorch 一起使用？我无法使用 model.to(device="GPU") 将模型移动到 GPU

Answer 1

PyTorch 目前不支持导入 onnx 模型。在撰写此答案时，它是一个开放的功能请求。

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

Answer 2

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