0
点赞
收藏
分享

微信扫一扫

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)


举报

相关推荐

0 条评论