我们不能运行导入到 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 的工作方式。