我有一个自定义dnn模型,该模型已在PyTorch中定义并接受了培训,我正尝试通过ONNX将其部署在用C#编写的应用程序中。
加载模型时,出现以下错误:
经过一些测试,我发现只有当我重新添加网络的上采样侧(使用转置卷积)时,这种情况才会发生。
Python API页面here显示了一个convolution_transpose
函数,但我在C#API中找不到一个函数。
谢谢!
答案 0 :(得分:0)
您可能需要更新CNTK,几天前已有新版本(2.6)可用,release notes中介绍了ONNX的更新:
对ConvolutionTranspose导出和导入进行大修。完全支持诸如output_shape,output_padding和pads之类的属性。