diff --git a/paddleslim/auto_compression/strategy_config.py b/paddleslim/auto_compression/strategy_config.py index c5fe72aacdcf9b1fe34d763a518bb7f70b2ba0a3..aa8b14ede101a46e03ea7c0696217e080d2ac368 100644 --- a/paddleslim/auto_compression/strategy_config.py +++ b/paddleslim/auto_compression/strategy_config.py @@ -68,7 +68,7 @@ class QuantAware(BaseStrategy): window_size=10000, moving_rate=0.9, for_tensorrt=False, - onnx_format=False, + onnx_format=True, is_full_quantize=False): """ Quantization Config. @@ -201,7 +201,7 @@ class QuantPost(BaseStrategy): activation_quantize_type='range_abs_max', simulate_activation_quant=False, skip_tensor_list=None, - onnx_format=False, + onnx_format=True, quantize_op_types=[ "conv2d", "depthwise_conv2d", "mul", "matmul", "matmul_v2" ],