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 6dcbf9e1e2..3097e1d82a 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: