Onnx精度转换 FP32-＞FP16

本文主要是介绍Onnx精度转换 FP32-＞FP16，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Onnx精度转换 FP32->FP16

1、依赖包

1. onnx
2. onnxmltools

2、转换

from onnxmltools.utils.float16_converter import convert_float_to_float16
from onnxmltools.utils import load_model,save_modelonnx_model = load_model("model.onnx")
fp16_model = convert_float_to_float16(onnx_model, keep_io_types=False)
save_model(fp16_model, "model-float16.onnx")

3、验证

import onnxprint(onnx.__version__)
onnx_model = onnx.load("model-float16.onnx")
try:onnx.checker.check_model(onnx_model)
except Exception:print("Model incorrect")
else:print("Model correct")

这篇关于Onnx精度转换 FP32-＞FP16的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---