oneflow_convert_tools

简体中文 | English

OneFlow 相关的模型转换工具

oneflow_onnx

简介

• OneFlow2ONNX 模型支持，支持 OneFlow 静态图模型转为 ONNX，可转换由 flow.checkpoint.save 方法保存下来的 OneFlow 模型，详情可以参考 OneFlow2ONNX 模型列表。
• OneFlow2ONNX 算子支持，目前稳定支持导出 ONNX Opset10，部分 OneFlow 算子支持更低的 ONNX Opset 转换，详情可以参考 OneFlow2ONNX 算子列表。

环境依赖

用户环境配置

python>=3.5
onnx>=1.8.0
onnx-simplifier>=0.3.3
onnxoptimizer>=0.2.5
onnxruntime>=1.6.0
oneflow>=0.5.0

安装

安装方式1

pip install oneflow_onnx

安装方式2

git clone https://github.com/Oneflow-Inc/oneflow_convert_tools
cd oneflow_onnx
python3 setup.py install

使用方法

请参考使用示例

相关文档

• OneFlow2ONNX模型列表
• OneFlow2ONNX算子列表
• 使用示例