前言
进行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)
或者使用静态量化见:文章来源:https://www.toymoban.com/news/detail-802834.html
模型量化(3):ONNX 模型的静态量化和动态量化 - 飞桨AI Studio星河社区文章来源地址https://www.toymoban.com/news/detail-802834.html
到了这里,关于[BUG] cannot import name ‘quantize_qat‘ from ‘onnxruntime.quantization‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!