如果用户对ONNX原始模型已经做了量化功能(以下简称QAT模型),但是该模型无法使用ATC工具转成适配昇腾AI处理器的离线模型,则需要借助AMCT提供的convert_qat_model接口,将该QAT模型适配成CANN量化模型格式。然后才能使用ATC工具,将CANN量化模型转成适配昇腾AI处理器的离线模型。转换约束如下:
适配原理如图1所示。蓝色部分为用户实现,灰色部分为用户调用AMCT提供的convert_qat_model实现,用户在ONNX QAT网络推理的代码中导入库,并在特定的位置调用相应API,即可实现适配功能。具体适配示例请参见获取更多样例>mobilenetv2,您也可以获取命令行方式的示例获取更多样例>cmd,快速体验其功能。
本示例演示了如何将ONNX的QAT量化模型通过AMCT适配为CANN量化模型格式。
import amct_onnx as amct
建议使用原始待量化的模型和测试集,在ONNX Runtime环境下推理,验证环境、推理脚本是否正常。
推荐执行该步骤,请确保原始模型可以完成推理且精度正常;执行该步骤时,可以使用部分测试集,减少运行时间。
user_do_inference(ori_model, test_data, test_iterations)
model_file = "./pre_model/mobilenet_v2_qat.onnx" save_path="./results/model" amct.convert_qat_model(model_file, save_path)
quant_model = './results/user_model_fake_quant_model.onnx' user_do_inference(quant_model, test_data, test_iterations)