Модель ONNX с ошибкой субоператора не связывает - PullRequest
0 голосов
/ 26 октября 2018

Я пытаюсь создать сеанс с моделью squeezenet ONNX, используя: session = winrt :: Windows :: AI :: MachineLearning :: LearningModelSession {модель, winrt :: Windows :: AI :: MachineLearning :: LearningModelDevice (deviceKind)};

У меня есть две версии squeezenet. Один из них имеет подуровень в начале, а другой - нет. Тот, что с 'Sub', выдает ошибку при выполнении вышеизложенного.

Есть идеи, что здесь происходит?

Ответы [ 2 ]

0 голосов
/ 03 декабря 2018

Какова форма ввода для squeezenet, который выходит из строя? Если вы опубликуете неисправную модель, мы поможем вам устранить ее.

2 ключевых момента, чтобы заставить его работать для сборки ОС 17763:

  1. Убедитесь, что вы используете ONNX версии 1.2 (opset 7)
  2. Убедитесь, что ваш ввод имеет правильную форму, ожидаемую моделью.

Какую ошибку вы получаете при создании LearningModelSession?

0 голосов
/ 29 ноября 2018

Наличие оператора 'Sub' на графике ONNX не должно влиять на то, можете ли вы запустить эту модель в Windows. Я думаю, что более важный вопрос - это версия ONNX (или версия набора операторов) и целевая версия Windows. Начиная с обновления в октябре 2018 года, машинное обучение Windows будет совместимо с ONNX версии 1.2.2 (https://github.com/onnx/onnx/releases/tag/v1.2.2).. Дважды проверьте, есть ли у вас модель 1.2.2 и SDK обновления в октябре 2018 года (10.0.17763.x)

...