AI Rookies

ONNX — 开放神经网络交换格式

事实

一种跨框架交换神经网络模型的开放格式。

人话

ONNX 是模型的旅行转换插头,这国充好电,换个插座也能通电。

用于跨框架部署,让模型少被单一框架锁死。

相关概念

Framework
ONNX 让不同框架训练出的模型更容易互通。

PyTorch
PyTorch 模型常导出为 ONNX,用于后续部署。

TensorFlow
TensorFlow 模型也可转换成 ONNX 格式。

Inference engine
推理引擎可读取 ONNX,把模型真正跑起来。