From 37c5222400451125964b49dc8118db4633307584 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Thu, 30 Mar 2023 13:24:25 +0800
Subject: [PATCH] Update strategy_config.py

---
 paddleslim/auto_compression/strategy_config.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddleslim/auto_compression/strategy_config.py b/paddleslim/auto_compression/strategy_config.py
index c5fe72aa..aa8b14ed 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"
                  ],
-- 
GitLab