bevformer / config.json
fangmingguo's picture
Update config.json
c8a2d3c verified
{
"output_name": "bevformer_tiny.axmodel",
"model_type": "ONNX",
"target_hardware": "AX650",
"npu_mode": "NPU3",
"quant": {
"input_configs": [
{
"tensor_name": "img",
"calibration_dataset": "./calibration/calibration_img.tar.gz",
"calibration_format": "Numpy",
"calibration_size": 64,
"calibration_mean": [0, 0, 0],
"calibration_std": [1, 1, 1]
},
{
"tensor_name": "lidar2img",
"calibration_dataset": "./calibration/calibration_lidar2img.tar.gz",
"calibration_format": "Numpy",
"calibration_size": 64
},
{
"tensor_name": "can_bus",
"calibration_dataset": "./calibration/calibration_can_bus.tar.gz",
"calibration_format": "Numpy",
"calibration_size": 64
},
{
"tensor_name": "prev_bev",
"calibration_dataset": "./calibration/calibration_prev_bev.tar.gz",
"calibration_format": "Numpy",
"calibration_size": 64
}
],
"calibration_method": "MSE",
"precision_analysis": true,
"precision_analysis_method": "EndToEnd",
"enable_smooth_quant": true,
"smooth_quant_threshold": 0.15,
"smooth_quant_strength": 0.5,
"refine_weight_threshold": 1e-6,
"conv_bias_data_type": "FP32",
"ln_scale_data_type": "FP32",
"layer_configs": [
{
"layer_name": "/pts_bbox_head/transformer/encoder/Reshape_10",
"data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/Tile_3",
"data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/MatMul",
"data_type": "FP32",
"output_data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/Squeeze",
"data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/Slice_10",
"data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/Slice_9",
"data_type": "FP32"
},
{
"layer_name": "/pts_bbox_head/transformer/encoder/Div_2",
"data_type": "FP32"
}
]
}
}