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

Запуск последней версии WinMLRunning, но с ошибкой:

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

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

ОС: 17758 инсайдерской превью
SDK: 17754 просмотр инсайдерской информации
VS: 15.8.4

WinML Runner
Графический процессор: Intel(R) UHD Graphics 620

=================================================================
Имя: tf2onnx
Автор: tf2onnx
Версия: 9223372036854775807
Домен:
Описание:
Путь: test3.onnx
Поддержка FP16: ложь

Информация о функции ввода:
Имя: вход:0
Особенность Вид: Поплавок

Информация о функции вывода:
Имя: InceptionV1/ Логиты / Предсказания /Reshape_1:0
Особенность Вид: Поплавок

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

Загрузка модели...[УСПЕХ]
Создание сеанса [FAILED]
Не найдено подходящего определения ядра для операции добавления (узел InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

=================================================================
Имя: tf2onnx
Автор: tf2onnx
Версия: 9223372036854775807
Домен:
Описание:
Путь: test3.onnx
Поддержка FP16: ложь

Информация о функции ввода:
Имя: вход:0
Особенность Вид: Поплавок

Информация о функции вывода:
Имя: InceptionV1/ Логиты / Предсказания /Reshape_1:0
Особенность Вид: Поплавок

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

Загрузка модели...[УСПЕХ]
Создание сеанса [FAILED]
Не найдено подходящего определения ядра для операции добавления (узел InceptionV1/InceptionV1/Conv2d_1a_7x7/BatchNorm/FusedBatchNorm)

1 ответ

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

Мы хотим: ai.onnx v7 для ONNX Operator Set версии 7, которая является минимальной версией, поддерживаемой API машинного обучения Windows.

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

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

Другие вопросы по тегам