onnx华为mdc

import amct_onnx as amct
config_file = './tmp/config.json'
skip_layers = []
batch_num = 1
amct.create_quant_config(config_file=config_file,
			 model_file=ori_model,
			 skip_layers=skip_layers,
			 batch_num=batch_num)
record_file = './tmp/record.txt'
modified_model = './tmp/modified_model.onnx'
amct.quantize_model(config_file=config_file,
                    ori_model=ori_model,
                    modified_onnx_file=modified_model,
                    record_file=record_file)
quant_model_path = './results/user_model'
amct.save_model(modified_onnx_file=modified_model,
                record_file=record_file,
                save_path=quant_model_path)
quant_model = './results/user_model_fake_quant_model.onnx'
user_do_inference(quant_model, test_data, test_iterations)
posted @ 2022-08-19 22:46  luoganttcc  阅读(50)  评论(0)    收藏  举报