temp fix erros in removing vars.

0x45f · Apr 16, 2022 · 2fd5fde · 2fd5fde
1 parent 4e27f67
commit 2fd5fde
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/autograd/primx.py b/python/paddle/autograd/primx.py
@@ -383,7 +383,8 @@ def _lower(block, reverse, update_var_list):
     for op_idx in reversed(ops_to_remove):
         block._remove_op(op_idx)
     for var_name in vars_to_remove:
-        block._remove_var(var_name)
+        # block._remove_var(var_name)
+        del block.vars[var_name]
 
     if update_var_list is not None:
         for i in range(len(update_var_list)):