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

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

diff --git a/paddleslim/quant/post_quant_hpo.py b/paddleslim/quant/post_quant_hpo.py
index c402ce4c..9ec0d0d5 100755
--- a/paddleslim/quant/post_quant_hpo.py
+++ b/paddleslim/quant/post_quant_hpo.py
@@ -77,7 +77,7 @@ class QuantConfig(object):
                  activation_bits=8,
                  weight_quantize_type='channel_wise_abs_max',
                  optimize_model=False,
-                 onnx_format=False,
+                 onnx_format=True,
                  is_use_cache_file=False,
                  cache_dir="./temp_post_training"):
         """QuantConfig init"""
@@ -358,7 +358,7 @@ def quant_post_hpo(
         batch_size=[10, 30],  ### uniform sample in list.
         batch_num=[10, 30],  ### uniform sample in list.
         optimize_model=False,
-        onnx_format=False,
+        onnx_format=True,
         is_use_cache_file=False,
         cache_dir="./temp_post_training",
         runcount_limit=30):
-- 
GitLab