ONNX支持哪些深度学习框架

1395
2024/3/21 21:05:25
栏目: 深度学习
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

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

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

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: 怎么查看onnx模型结构权重