我已经使用Pytorch为CIFAR-10建立了模型,并将其保存为ONNX文件。
但是看来我无法从CNTK加载它。
我已经从相同的源代码加载了另一个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.)
答案 0 :(得分:1)
该模型似乎无法加载到CNTK中。 CNTK对导出(保存)到ONNX具有良好的支持,对于某些操作来说导入(加载)可能会遇到问题。
CNTK的开发被冻结了,您使用它的动机是什么?
现在推荐的方法是使用ONNX运行时https://github.com/microsoft/onnxruntime进行推理,它对ONNX具有一流的支持。