模型预测结果数据类型错误

通过以下yaml将模型转换成bin,在预测时发现 result = models[0].forward(input_tensor) result[0].buffer 类型为int32。如何保证预测结果为float32?

model_parameters

onnx_model: ‘./model.onnx’

march: “bayes-e”

layer_out_dump: False

working_dir: ‘model_output’

output_model_file_prefix: ‘nn’

remove_node_type: “Dequantize”

input_parameters:

input_name: “”

input_type_rt: ‘nv12’

input_type_train: ‘rgb’

input_layout_train: ‘NCHW’

input_shape: ‘’

norm_type: ‘data_scale’

mean_value: ‘’

scale_value: 0.003921568627451

calibration_parameters:

cal_data_dir: ‘./calibration_data_rgb_f32’

cal_data_type: ‘float32’

calibration_type: ‘default’

compiler_parameters:

compile_mode: ‘latency’

debug: False

optimize_level: ‘O3’

你好, 这个是配置了移除反量化节点的参数导致的,在手册的PTQ流程介绍中,有相关的说明:https://developer.d-robotics.cc/api/v1/fileData/x5\_doc-v126cn/oe\_mapper/source/ptq/ptq\_tool/hb\_mapper/hb\_mapper\_makertbin.html