CNTK:“无法根据输入和新形状尺寸来计算推断的尺寸。”

时间:2019-06-19 22:58:02

标签: pytorch cntk onnx

我已经使用Pytorch为CIFAR-10建立了模型,并将其保存为ONNX文件。

但是看来我无法从CNTK加载它。

Architecture

我已经从相同的源代码加载了另一个ONNX文件(错误),因此依赖项看起来还可以。当我调用Function.Load()

时出现问题
var deviceDescriptor = DeviceDescriptor.CPUDevice; ;
var function = Function.Load(ONNX_PATH, deviceDescriptor, ModelFormat.ONNX);

我收到此异常(未处理的异常):

   System.ApplicationException : 'Reshape: inferred dimension cannot be calculated from input and new shape size.

    [CALL STACK]
    - CNTK::TrainingParameterSchedule::  GetMinibatchSize
    - CNTK::  XavierInitializer (x6)
    - CNTK::Function::Load
    - CSharp_CNTK_Function__Load__SWIG_0
    - 00007FFB0C41C307 (SymFromAddr() error: Le module spécifié est introuvable.)

1 个答案:

答案 0 :(得分:1)

该模型似乎无法加载到CNTK中。 CNTK对导出(保存)到ONNX具有良好的支持,对于某些操作来说导入(加载)可能会遇到问题。

CNTK的开发被冻结了,您使用它的动机是什么?

现在推荐的方法是使用ONNX运行时https://github.com/microsoft/onnxruntime进行推理,它对ONNX具有一流的支持。