| { | |
| "input": "/opt/rzyang/Github/Magnetar/todos/work/20260706-124435-diarizen/export_cnn/cnn_features.onnx", | |
| "output_dir": "/opt/rzyang/Github/Magnetar/todos/work/20260706-124435-diarizen/compile_cnn_u16_full", | |
| "output_name": "cnn_features.axmodel", | |
| "work_dir": "/opt/rzyang/Github/Magnetar/todos/work/20260706-124435-diarizen/compile_cnn_u16_full/work_dir", | |
| "model_type": "ONNX", | |
| "target_hardware": "AX650", | |
| "npu_mode": "NPU3", | |
| "input_shapes": "waveform_ln:1x64000", | |
| "onnx_opt": { | |
| "model_check": true, | |
| "disable_transformation_check": true | |
| }, | |
| "quant": { | |
| "input_configs": [ | |
| { | |
| "tensor_name": "waveform_ln", | |
| "calibration_dataset": "/opt/rzyang/Github/Magnetar/todos/work/20260706-124435-diarizen/compile_cnn/calibration/calib.tar", | |
| "calibration_format": "Numpy", | |
| "calibration_size": 1, | |
| "calibration_mean": [], | |
| "calibration_std": [] | |
| } | |
| ], | |
| "calibration_method": "Percentile", | |
| "precision_analysis": false, | |
| "highest_mix_precision": false, | |
| "layer_configs": [ | |
| { | |
| "op_types": [ | |
| "Conv", | |
| "Mul", | |
| "Add", | |
| "Transpose", | |
| "ReduceMean", | |
| "Div", | |
| "Sub", | |
| "Pow", | |
| "Sqrt", | |
| "Erf", | |
| "Unsqueeze", | |
| "Gather", | |
| "Reshape", | |
| "Squeeze", | |
| "Concat", | |
| "Slice", | |
| "Softmax", | |
| "Sigmoid", | |
| "MatMul", | |
| "Gemm", | |
| "Split", | |
| "LayerNormalization" | |
| ], | |
| "data_type": "U16" | |
| } | |
| ] | |
| }, | |
| "input_processors": [ | |
| { | |
| "tensor_name": "waveform_ln", | |
| "tensor_layout": "NCHW", | |
| "src_layout": "NCHW", | |
| "src_dtype": "FP32", | |
| "mean": [], | |
| "std": [] | |
| } | |
| ], | |
| "compiler": { | |
| "check": 0, | |
| "npu_perf": true | |
| } | |
| } |