温馨提示×

ONNX支持哪些深度学习框架

小樊
55
2024-03-26 21:05:25
栏目: 深度学习

ONNX(Open Neural Network Exchange)支持以下深度学习框架:

  1. PyTorch
  2. TensorFlow
  3. MXNet
  4. Caffe2

此外,ONNX还支持其他一些框架的模型转换和互操作性,包括但不限于Keras、CNTK和Scikit-learn。通过ONNX,用户可以在不同框架之间共享和转移训练好的模型,从而更方便地部署模型或进行模型融合。

0