onnx - 子操作员错误的 ONNX 模型不绑定

Question

我正在尝试使用squeezenet ONNX模型创建会话： session = winrt::Windows::AI::MachineLearning::LearningModelSession{ model, winrt::Windows::AI::MachineLearning::LearningModelDevice(deviceKind) };

我有两个版本的squeezenet。一个在开头有一个“子”层，另一个没有。执行上述操作时，带有“Sub”的会引发错误。

有什么想法吗？

Answer 1

失败的squeezenet输入的形状是什么？如果您发布失败的模型，我们可以帮助您排除故障。

使其适用于 OS build 17763 的 2 个关键事项：

1. 确保您使用的是 ONNX 版本 1.2（opset 7）
2. 确保您的输入具有模型所期望的正确形状。

创建 LearningModelSession 时遇到什么错误？

Answer 2

ONNX 图中“Sub”运算符的存在不应影响您是否可以在 Windows 上运行该模型。我认为更重要的问题是 ONNX 版本（或算子集版本）和目标 Windows 版本。从 2018 年 10 月更新开始，windows 机器学习将兼容 ONNX 1.2.2 版本（https://github.com/onnx/onnx/releases/tag/v1.2.2）。仔细检查您是否拥有模型 1.2.2 和 2018 年 10 月更新 SDK (10.0.17763.x)