windows-machine-learning - "Не найдено подходящего определения ядра для op Add" - PullRequest
Новая
       29

windows-machine-learning - "Не найдено подходящего определения ядра для op Add"

0 голосов
/ 17 сентября 2018

Выполняется последняя WinMLRunning , но появляется ошибка:

Не найдено подходящего определения ядра для op Add (узел InceptionV1 / InceptionV1 / Conv2d_1a_7x7 / BatchNorm / FusedBatchNorm)

Я преобразовал замороженную модель начального вывода в ONNX, используя VS или mmdnn.Я получаю похожую ошибку при написании нового рабочего стола Win32 WinML с нуля.

OS: 17758 insider preview
SDK : 17754 insider preview
VS: 15.8.4

WinML Runner
GPU: Intel(R) UHD Graphics 620

=================================================================
Name: tf2onnx
Author: tf2onnx
Version: 9223372036854775807
Domain:
Description:
Path: test3.onnx
Support FP16: false

Input Feature Info:
Name: input:0
Feature Kind: Float

Output Feature Info:
Name: InceptionV1/Logits/Predictions/Reshape_1:0
Feature Kind: Float

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

Loading model...[SUCCESS]
Creating session [FAILED]
No suitable kernel definition found for op Add (node InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

=================================================================
Name: tf2onnx
Author: tf2onnx
Version: 9223372036854775807
Domain:
Description:
Path: test3.onnx
Support FP16: false

Input Feature Info:
Name: input:0
Feature Kind: Float

Output Feature Info:
Name: InceptionV1/Logits/Predictions/Reshape_1:0
Feature Kind: Float

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

Loading model...[SUCCESS]
Creating session [FAILED]
No suitable kernel definition found for op Add (node InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

1 Ответ

0 голосов
/ 26 сентября 2018

Пожалуйста, убедитесь, что установлена ​​версия вашего оператора.Чтобы проверить версию, откройте модель с помощью Netron , щелкните узел ввода и проверьте поле Imports.

Мы хотим: ai.onnx v7 для набора операторов ONNXВерсия 7, которая является минимальной версией, поддерживаемой Windows Machine Learning API.

Попробуйте использовать инструмент преобразования tf2onnx для преобразования из tenorflow.Вам также может потребоваться обновить пакет ONNX Python.Мы поддерживаем ONNX 1.2.2 или выше : https://github.com/onnx/tensorflow-onnx

Эта проблема была решена здесь: https://github.com/Microsoft/Windows-Machine-Learning/issues/50

...