发布于: Nov 16, 2017

开放神经网络交换 (ONNX) 是一种用于对深度学习模型进行编码的开源格式。现在提供了 ONNX-MXNet 开源 Python 软件包,可供开发人员使用 PyTorch、CNTK 或 Caffe2 之类的其他框架来构建和训练模型,并将这些模型导入到 Apache MXNet 中,以便使用 MXNet 高度优化的引擎运行这些模型来进行推理。

ONNX 可定义神经网络计算图的格式以及图中所用大量运算符的格式。随着越来越多的框架和硬件供应商支持 ONNX,从事深度学习的开发人员可以轻松地在框架间移动,选择最适合当前任务的框架。

AWS 正在与 Facebook、Microsoft 和深度学习社区合作,深入开发 ONNX 格式,让深度学习从业人员都可以访问和使用它。

利用 MXNet,用户可以轻松地导入模型并运行它们进行推理。通过 ONNX-MXNet GitHub 存储库中的本示例来试用 ONNX。另外可以查看 ONNX,深入讨论如何对网络图和运算符进行编码。欢迎您投稿!