Use the specificed scope in post quant, test=develop (#25384)

5c8e7995 · cc · GitHub · eb3173e2 · 5c8e7995
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py ...d/contrib/slim/quantization/post_training_quantization.py +2 -1

未找到文件。
--- a/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py
+++ b/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py
@@ -342,7 +342,8 @@ class PostTrainingQuantization(object):
            self._executor.run(program=self._program,
                               feed=data,
                               fetch_list=self._fetch_list,
-                               return_numpy=False)
+                               return_numpy=False,
+                               scope=self._scope)
            if self._algo == "KL":
                self._sample_data(batch_id)
            else: