1

运行最新的 WinMLRunning但出现错误:

没有为 op Add 找到合适的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

我使用 VS 或 mmdnn 将自定义推理冻结 Inception 模型转换为 ONNX。从头开始编写新的 Win32 WinML 桌面应用程序时,我遇到了类似的错误。

OS:17758内幕预览
SDK : 17754 内幕预览
VS:15.8.4

WinML 亚军
GPU:英特尔(R) 超高清显卡 620

==================================================== ================
名称:tf2onnx
作者:tf2onnx
版本号:9223372036854775807
领域:
描述:
路径:test3.onnx
支持FP16:假

输入特征信息:
名称:输入:0
特征种类:浮动

输出特征信息:
名称:InceptionV1/Logits/Predictions/Reshape_1:0
特征种类:浮动

==================================================== ================

正在加载模型...[成功]
创建会话 [失败]
没有为 op Add 找到合适的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

==================================================== ================
名称:tf2onnx
作者:tf2onnx
版本号:9223372036854775807
领域:
描述:
路径:test3.onnx
支持FP16:假

输入特征信息:
名称:输入:0
特征种类:浮动

输出特征信息:
名称:InceptionV1/Logits/Predictions/Reshape_1:0
特征种类:浮动

==================================================== ================

正在加载模型...[成功]
创建会话 [失败]
没有为 op Add 找到合适的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)
4

1 回答 1

3

请确保您的操作员设置版本是最新的。要检查版本,请使用Netron打开模型,单击输入节点并检查 Imports 字段。

我们想要:适用于 ONNX Operator Set Version 7 的ai.onnx v7,这是 Windows 机器学习 API 支持的最低版本。

尝试使用 tf2onnx 转换器工具从 tensorflow 转换。您可能还需要更新您的 ONNX python 包。我们支持ONNX 1.2.2 或更高版本https ://github.com/onnx/tensorflow-onnx

此问题已在此处解决:https ://github.com/Microsoft/Windows-Machine-Learning/issues/50

于 2018-09-26T18:29:00.943 回答