run ()
功能说明
运行量化算法,初始化OnnxCalibrator后通过run()函数来执行量化。
调用示例
from msmodelslim.onnx.squant_ptq import OnnxCalibrator, QuantConfig quant_config = QuantConfig(disable_names=[], quant_mode=0, amp_num=0) output_model_path="/home/xxx/Resnet50/resnet50_quant.onnx" #根据实际情况配置 calibrator = Calibrator(input_model_path, quant_config) calibrator.run() calibrator.export_quant_onnx(output_model_path)
父主题: squant_ptq接口