fangmingguo commited on
Commit
d702276
·
verified ·
1 Parent(s): 248c766

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +131 -0
config.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input": "/data/huichiwen/dev/cp/onnx_model/pointpillars.onnx",
3
+ "output_dir": "./output",
4
+ "output_name": "centerpoint.axmodel",
5
+ "model_type": "ONNX",
6
+ "target_hardware": "AX650",
7
+ "npu_mode": "NPU3",
8
+
9
+ "input_shapes": "input.1:1x10x30000x20;indices_input:1x30000x2",
10
+
11
+ "quant": {
12
+ "input_configs": [
13
+ {
14
+ "tensor_name": "input.1",
15
+ "calibration_dataset": "/data/huichiwen/dev/cp/quant_data/features.tar.gz",
16
+ "calibration_format": "Numpy",
17
+ "calibration_size": 32
18
+ },
19
+ {
20
+ "tensor_name": "indices_input",
21
+ "calibration_dataset": "/data/huichiwen/dev/cp/quant_data/indices.tar.gz",
22
+ "calibration_format": "Numpy",
23
+ "calibration_size": 32
24
+ }
25
+ ],
26
+ "calibration_method": "MinMax",
27
+ "enable_smooth_quant": false,
28
+ "smooth_quant_strength": 0.5,
29
+ "layer_configs": [
30
+ {
31
+ "layer_names": [
32
+ "pfe_/pfn_layers.0/linear/MatMul",
33
+ "pfe_/pfn_layers.0/norm/BatchNormalization",
34
+ "pfe_/pfn_layers.0/Relu",
35
+ "pfe_/pfn_layers.0/ReduceMax",
36
+ // "pfe_/pfn_layers.0/Tile",
37
+ // "pfe_/pfn_layers.0/Concat",
38
+ "pfe_/pfn_layers.1/linear/MatMul",
39
+ "pfe_/pfn_layers.1/norm/BatchNormalization",
40
+ "pfe_/pfn_layers.1/Relu",
41
+ "pfe_/pfn_layers.1/ReduceMax",
42
+ "ScatterND_1"
43
+
44
+ ],
45
+ "data_type": "FP32",
46
+ "output_data_type": "FP32"
47
+ },
48
+ {
49
+ "layer_names": [
50
+ "/neck/blocks.0/blocks.0.1/Conv",
51
+ "/neck/blocks.0/blocks.0.3/Relu",
52
+ "/neck/blocks.0/blocks.0.4/Conv",
53
+ "/neck/blocks.0/blocks.0.6/Relu",
54
+ "/neck/blocks.0/blocks.0.7/Conv",
55
+ "/neck/blocks.0/blocks.0.9/Relu",
56
+ "/neck/blocks.0/blocks.0.10/Conv",
57
+ "/neck/blocks.0/blocks.0.12/Relu",
58
+ "/neck/deblocks.0/deblocks.0.0/Conv",
59
+ "/neck/blocks.1/blocks.1.1/Conv",
60
+ "/neck/deblocks.0/deblocks.0.2/Relu",
61
+ "/neck/blocks.1/blocks.1.3/Relu",
62
+ "/neck/blocks.1/blocks.1.4/Conv",
63
+ "/neck/blocks.1/blocks.1.6/Relu",
64
+ "/neck/blocks.1/blocks.1.7/Conv",
65
+ "/neck/blocks.1/blocks.1.9/Relu",
66
+ "/neck/blocks.1/blocks.1.10/Conv",
67
+ "/neck/blocks.1/blocks.1.12/Relu",
68
+ "/neck/blocks.1/blocks.1.13/Conv",
69
+ "/neck/blocks.1/blocks.1.15/Relu",
70
+ "/neck/blocks.1/blocks.1.16/Conv",
71
+ "/neck/blocks.1/blocks.1.18/Relu",
72
+ "/neck/deblocks.1/deblocks.1.0/Conv",
73
+ "/neck/deblocks.1/deblocks.1.2/Relu",
74
+ "/neck/blocks.2/blocks.2.1/Conv",
75
+ "/neck/blocks.2/blocks.2.3/Relu",
76
+ "/neck/blocks.2/blocks.2.4/Conv",
77
+ "/neck/blocks.2/blocks.2.6/Relu",
78
+ "/neck/blocks.2/blocks.2.7/Conv",
79
+ "/neck/blocks.2/blocks.2.9/Relu",
80
+ "/neck/blocks.2/blocks.2.10/Conv",
81
+ "/neck/blocks.2/blocks.2.12/Relu",
82
+ "/neck/blocks.2/blocks.2.13/Conv",
83
+ "/neck/blocks.2/blocks.2.15/Relu",
84
+ "/neck/blocks.2/blocks.2.16/Conv",
85
+ "/neck/blocks.2/blocks.2.18/Relu"
86
+ ],
87
+ "data_type": "FP32",
88
+ "output_data_type": "FP32"
89
+ },
90
+ {
91
+ "layer_names": [
92
+ "/Exp",
93
+ "/Sigmoid",
94
+ "/Exp_1",
95
+ "/Sigmoid_1",
96
+ "/Exp_2",
97
+ "/Sigmoid_2",
98
+ "/Exp_3",
99
+ "/Sigmoid_3",
100
+ "/Exp_4",
101
+ "/Sigmoid_4",
102
+ "/Exp_5",
103
+ "/Sigmoid_5",
104
+ "/ReduceMax",
105
+ "/ReduceMax_1",
106
+ "/ReduceMax_2",
107
+ "/ReduceMax_3",
108
+ "/ReduceMax_4",
109
+ "/ReduceMax_5"
110
+
111
+ ],
112
+ "data_type": "FP32",
113
+ "output_data_type": "FP32"
114
+ }
115
+ ]
116
+ },
117
+ "input_processors": [
118
+ {
119
+ "tensor_name": "input.1",
120
+ "src_dtype": "FP32"
121
+ },
122
+ {
123
+ "tensor_name": "indices_input",
124
+ "src_dtype": "S32"
125
+ }
126
+ ],
127
+
128
+ "compiler": {
129
+ "check": 0
130
+ }
131
+ }