我正在尝试使用squeezenet ONNX模型创建会话: session = winrt :: Windows :: AI :: MachineLearning :: LearningModelSession {模型,winrt :: Windows :: AI :: MachineLearning :: LearningModelDevice(deviceKind)};
我有两个版本的squeezenet。一个在开始时具有“ Sub”层,而另一个则没有。执行上述操作时,带有“ Sub”的对象将引发错误。
有什么想法吗?
答案 0 :(得分:3)
出现故障的squeezenet输入的形状是什么?如果您发布失败的模型,我们可以帮助您进行故障排除。
使其适用于OS内部版本17763的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)