0

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

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

建筑学

我已经从相同的源代码(错误地)加载了另一个 ONNX 文件,所以依赖项看起来不错。当我调用 Function.Load() 时出现问题

var deviceDescriptor = DeviceDescriptor.CPUDevice; ;
var function = Function.Load(ONNX_PATH, deviceDescriptor, ModelFormat.ONNX);

我得到这个异常(未处理的异常):

   System.ApplicationException:'重塑:推断的尺寸不能从输入和新的形状大小计算。

    [呼叫堆栈]
    - CNTK::TrainingParameterSchedule::GetMinibatchSize
    - CNTK:: XavierInitializer (x6)
    - CNTK::函数::加载
    - CSharp_CNTK_Function__Load__SWIG_0
    - 00007FFB0C41C307(SymFromAddr() 错误:Le 模块 spécifié© est introuvable。)
4

1 回答 1

1

看起来这个模型无法加载到 CNTK 中。CNTK 对导出(保存)到 ONNX 有很好的支持,导入(加载)对于某些操作可能会出现问题。

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

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

于 2019-06-19T23:38:43.060 回答