"No suitable kernel definition found for op Add"



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

找不到适合 op Add 的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm(

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

OS:17758 预览体验成员预览版 SDK : 17754 预览体验成员 VS:15.8.4 赢机器学习运行程序 GPU: 英特尔® 超高清显卡 620 ================================================================= 名称: tf2onnx 作者: tf2onnx 版本: 9223372036854775807 域: 描述: 路径:测试3.onnx 支持 FP16:假 输入功能信息: 名称:输入:0 功能种类:浮点型 输出功能信息: 名称:InceptionV1/Logits/Predictions/Reshape_1:0 功能种类:浮点型 ================================================================= 加载模型...[成功] 创建会话 [失败] 找不到适合 op Add 的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm( ================================================================= 名称: tf2onnx 作者: tf2onnx 版本: 9223372036854775807 域: 描述: 路径:测试3.onnx 支持 FP16:假 输入功能信息: 名称:输入:0 功能种类:浮点型 输出功能信息: 名称:InceptionV1/Logits/Predictions/Reshape_1:0 功能种类:浮点型 ================================================================= 加载模型...[成功] 创建会话 [失败] 找不到适合 op Add 的内核定义(节点 InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm(

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

我们需要:ai.onnx v7for ONNX Operator Set Version 7,这是 Windows Machine Learning API 支持的最低版本。

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

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

最新更新