MMdnn是微软开源的一套帮助用户在不同深度学习框架之间进行互操作的工具,包括模型转换和可视化。目前支持在Caffe、Keras、MXNet、Tensorflow、CNTK、PyTorch和CoreML等框架之间进行模型转换。
MMdnn是一个用于转换、可视化和诊断深度神经网络模型的综合跨框架解决方案。MMdnn中的「MM」代表模型管理,「dnn」是「deepneuralnetwork」深度神经网络的缩写。
MMdnn可将一个框架训练的DNN模型转换到成其他框架可用的模型。主要特性如下:
模型文件转换器,转换DNN模型使其适合不同框架;
模型代码块生成器,为框架生成适合的训练或推断代码块;
模型可视化,可视化框架的DNN网络架构和参数;
模型兼容性测试(正在进行)。
评论