本文介绍: cannot import name ‘quantize_qat’ from ‘onnxruntime.quantization’

前言

进行onnx模型量化出现了这个错误

解决:

卸载重装onnx和onnxruntime,会自动安装最新,由于版本更新取消了quantize_qat函数

改为以下代码:

from onnxruntime.quantization import quantize_dynamic, QuantType,quantize_static

model_fp32 = 'yolov5s.onnx'
model_quant = 'yolov5s_dyn.onnx'
quantized_model = quantize_dynamic(model_fp32, model_quant)

或者使用静态量化见:

模型量化(3):ONNX 模型的静态量化和动态量化 – 飞桨AI Studio星河社区

原文地址:https://blog.csdn.net/qq_55542491/article/details/135606009

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。

如若转载,请注明出处:http://www.7code.cn/show_58586.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱:suwngjj01@126.com进行投诉反馈,一经查实,立即删除!

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注