fix ac import ptq_hpo (PaddlePaddle#1170)

lizexu123 · Jun 14, 2022 · 3ae057a · 3ae057a
1 parent 1b498f1
commit 3ae057a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py
@@ -35,7 +35,7 @@
 
 try:
     if platform.system().lower() == 'linux':
-        from ..quant import quant_post_hpo
+        from ..quant import post_quant_hpo
 except Exception as e:
     _logger.warning(e)
 
@@ -409,7 +409,7 @@ def compress(self):
 
         if strategy == 'ptq_hpo' and config.max_quant_count == 1 and platform.system(
         ).lower() == 'linux':
-            ptq_loss = quant_post_hpo.g_min_emd_loss
+            ptq_loss = post_quant_hpo.g_min_emd_loss
 
             final_quant_config = get_final_quant_config(ptq_loss)
             if final_quant_config is not None:
@@ -468,7 +468,7 @@ def single_strategy_compress(self, strategy, config, strategy_idx):
                 raise NotImplementedError(
                     "post-quant-hpo is not support in system other than linux")
 
-            quant_post_hpo.quant_post_hpo(
+            post_quant_hpo.quant_post_hpo(
                 self._exe,
                 self._places,
                 model_dir=self.model_dir,

diff --git a/paddleslim/quant/__init__.py b/paddleslim/quant/__init__.py
@@ -34,7 +34,7 @@
     from .quanter import quant_post, quant_post_only_weight
     from .quant_aware_with_infermodel import quant_aware_with_infermodel, export_quant_infermodel
     if platform.system().lower() == 'linux':
-        from .quant_post_hpo import quant_post_hpo
+        from .post_quant_hpo import quant_post_hpo
     else:
         _logger.warning(
             "post-quant-hpo is not support in system other than linux")

diff --git a/paddleslim/quant/quant_post_hpo.py → paddleslim/quant/post_quant_hpo.py b/paddleslim/quant/quant_post_hpo.py → paddleslim/quant/post_quant_hpo.py