Skip to content

Commit cceceed

Browse files
intel-zhangyizehao-intel
authored andcommitted
Revert "remove op-wise cfgs for testing. (#1521)" (#202)
Signed-off-by: intel-zhangyi <[email protected]> Signed-off-by: zehao-intel <[email protected]>
1 parent 68a0783 commit cceceed

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed

examples/tensorflow/image_recognition/tensorflow_models/quantization/ptq/densenet121.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ quantization: # optional. tuning constrai
3838
algorithm: minmax
3939
weight:
4040
granularity: per_channel
41+
op_wise: {
42+
'densenet121/MaxPool2D/MaxPool': {
43+
'activation': {'dtype': ['fp32']}
44+
},
45+
'densenet121/transition_block[1-3]/AvgPool2D/AvgPool': {
46+
'activation': {'dtype': ['fp32']},
47+
}
48+
}
4149

4250
evaluation: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.
4351
accuracy: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.

examples/tensorflow/image_recognition/tensorflow_models/quantization/ptq/densenet161.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ quantization: # optional. tuning constrai
3838
algorithm: minmax
3939
weight:
4040
granularity: per_channel
41+
op_wise: {
42+
'densenet161/MaxPool2D/MaxPool': {
43+
'activation': {'dtype': ['fp32']}
44+
},
45+
'densenet161/transition_block[1-3]/AvgPool2D/AvgPool': {
46+
'activation': {'dtype': ['fp32']},
47+
}
48+
}
4149

4250
evaluation: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.
4351
accuracy: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.

examples/tensorflow/image_recognition/tensorflow_models/quantization/ptq/densenet169.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ quantization: # optional. tuning constrai
3838
algorithm: minmax
3939
weight:
4040
granularity: per_channel
41+
op_wise: {
42+
'densenet169/MaxPool2D/MaxPool': {
43+
'activation': {'dtype': ['fp32']}
44+
},
45+
'densenet169/transition_block[1-3]/AvgPool2D/AvgPool': {
46+
'activation': {'dtype': ['fp32']},
47+
}
48+
}
4149

4250
evaluation: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.
4351
accuracy: # optional. required if user doesn't provide eval_func in neural_compressor.Quantization.

0 commit comments

Comments
 (0)