accuarcy-debug.zip

# 导入debug模块 import horizon_nn.debug as dbg # 导入log日志模块 import logging # 若verbose=True时，需要先设置log level为INFO logging.getLogger().setLevel(logging.INFO) # 获取节点量化敏感度 # node_message = dbg.get_sensitivity_of_nodes( # model_or_file='ai_benchmark/j5/ptq/tools/python_tools/dsgn2_cal_calibrated_model.onnx', # metrics=['cosine-similarity', 'mse'], # calibrated_data='ai_benchmark/j5/ptq/tools/python_tools/calibration_data/', # output_node=None, # node_type='node', # data_num=None, # verbose=True, # interested_nodes=None) # nodes = list(node_message.keys()) dbg.plot_acc_error(save_dir='ai_benchmark/j5/ptq/tools/python_tools/', calibrated_data='ai_benchmark/j5/ptq/tools/python_tools/calibration_data/', model_or_file='ai_benchmark/j5/ptq/tools/python_tools/dsgn2_cal_calibrated_model.onnx', non_quantize_node=[['MatMul_368_reshape_output'], ['MatMul_368_reshape_output', 'Div_374_reciprocal']], metric='cosine-similarity', average_mode=False)