Поддерживает ли CNTK C # API ConvTranspose? - PullRequest
0 голосов
/ 18 сентября 2018

У меня есть пользовательская модель dnn, которую я определил и обучил в PyTorch, которую я пытаюсь развернуть в приложении, написанном на C # через ONNX.

Когда я загружаю модель, у меня появляется следующая ошибка:

image CNTK import failure">

После некоторого тестирования я обнаружил, что это происходит только тогда, когда я снова добавляю обратно сторону дискретизации сети (которая использует транспонированную свертку).

Страницы API Python здесь показывают функцию convolution_transpose, но я не могу найти ее в C # API.

Спасибо!

1 Ответ

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

Вам может потребоваться обновить CNTK, новая версия (2.6) доступна несколько дней назад, и обновления ONNX описаны в заметках о выпуске :

Капитальный ремонт ConvolutionTranspose экспорт и импорт. Такие атрибуты, как> output_shape, output_padding и pads полностью поддерживаются.

...