带有子运算符错误的ONNX模型未绑定

时间:2018-10-26 16:18:00

标签: onnx windows-machine-learning

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

我有两个版本的squeezenet。一个在开始时具有“ Sub”层,而另一个则没有。执行上述操作时,带有“ Sub”的对象将引发错误。

有什么想法吗?

2 个答案:

答案 0 :(得分:3)

出现故障的squeezenet输入的形状是什么?如果您发布失败的模型,我们可以帮助您进行故障排除。

使其适用于OS内部版本17763的2个关键事项:

  1. 确保您使用的是ONNX 1.2版(操作集7)
  2. 确保您的输入具有模型期望的正确形状。

创建LearningModelSession时遇到什么错误?

答案 1 :(得分:1)

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)