clear tmp var in ptq (PaddlePaddle#48660)

lxsbupt · Dec 6, 2022 · dbe0595 · dbe0595
1 parent 426bb6e
commit dbe0595
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py b/python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py
@@ -692,11 +692,10 @@ def _reset_activation_persistable(self):
         '''
         Reset activations to be not persistable.
         '''
-        to_erase = []
         for var in self._program.list_vars():
             if var.name in self._quantized_act_var_name:
                 var.persistable = False
-                to_erase.append(var.name)
+                self._scope.find_var(var.name).get_tensor()._clear()
 
     def _sampling(self):
         '''