ONNX

维基百科,自由的百科全书
跳到导航 跳到搜索

ONNX(英語:Open Neural Network Exchange)是一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型。它使得不同的人工智能框架(如Pytorch、MXNet)可以采用相同格式存储模型数据并交互。 ONNX的规范及代码主要由微软亚马逊FacebookIBM等公司共同开发,以开放源代码的方式托管在Github上。[1][2][3]  目前官方支持加载ONNX模型并进行推理的深度学习框架有: Caffe2, PyTorch, MXNet,ML.NET,TensorRT 和 Microsoft CNTK,并且 TensorFlow 也非官方的支持ONNX。

历史[编辑]

ONNX的第一个正式版本(v1.0)发布于2017年12月。

参考文献[编辑]

  1. ^ onnx: Open Neural Network Exchange, onnx, [2017-09-11], (原始内容存档于2017-09-09) 
  2. ^ Microsoft, Facebook Partner on AI Framework Interoperability. 
  3. ^ Model Server for Apache MXNet introduces ONNX support and Amazon CloudWatch integration. [8 February 2018]. (原始内容存档于2019-02-17).