From 5c8e79956e78494087f060f41208fa9ec938f436 Mon Sep 17 00:00:00 2001 From: cc <52520497+juncaipeng@users.noreply.github.com> Date: Tue, 7 Jul 2020 12:33:47 +0800 Subject: [PATCH] Use the specificed scope in post quant, test=develop (#25384) --- .../contrib/slim/quantization/post_training_quantization.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py b/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py index 6dcbf9e1e27..3097e1d82a9 100644 --- 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: -- GitLab