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 6dcbf9e1e278afd574709c44948f2a9b8b97096b..3097e1d82a9cb5e096efa3913ea6a06bff557c94 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: