Question

我知道的唯一保存训练模型的函数是 trainer.SaveCheckpoint，它可以保存 CNTK 模型，但我找不到如何在 C# 中将模型保存为 ONNX 格式

在此处的文档站点 https://docs.microsoft.com/en-us/cognitive-toolkit/serialization

我只能找到将其另存为ONNX的python方法

z.save("myModel.onnx", format=C.ModelFormat.ONNX)

但这在 C# 中不起作用

Answer 1

您可以考虑使用 microsoft 的 MMdnn 将您的 CNTK 模型转换为 onnx。

MMdnn 是一个全面的跨框架工具，用于转换、可视化和诊断深度学习 (DL) 模型。 “MM”代表模型管理，“dnn”是深度神经网络的缩写。

Answer 2

.NET Managed library 的

Function 对象有一个私有方法 _Save，其签名如下：

private void _Save(string filepath, ModelFormat format)

你可以通过反射来执行它：

string cntkFilePath = "myModel.model";
string onnxFilePath = "myModel.onnx";

Function model = Function.Load(cntkFilePath, DeviceDescriptor.CPUDevice, ModelFormat.CNTKv2);
MethodInfo saveMethod = typeof(Function).GetMethod(
    "_Save",
    BindingFlags.NonPublic | BindingFlags.Instance,
    null, 
    new[] { typeof(string), typeof(ModelFormat) },
    null);

saveMethod?.Invoke(model, new object[] { onnxFilePath, ModelFormat.ONNX });

如何在 C# 中将 CNTK 模型保存为 ONNX

2 个答案: