通过以下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’