ONNX介紹

ONNX是一種針對機器學習所設計的開放式的文件格式,用於存儲訓練好的模型。它使得不一樣的人工智能框架(如Pytorch, MXNet)能夠採用相同格式存儲模型數據並交互。 ONNX的規範及代碼主要由微軟,亞馬遜 ,Facebook 和 IBM 等公司共同開發,以開放源代碼的方式託管在Github上。目前官方支持加載ONNX模型並進行推理的深度學習框架有: Caffe2, PyTorch, MXNet,ML.NET,TensorRT 和 Microsoft CNTK,而且 TensorFlow 也非官方的支持ONNX。
 
開放、標準、可移植
相關文章
相關標籤/搜索