[Eager]Fix clear_gradient bug in optimizer (#41658)

PaddlePaddle · Apr 12, 2022 · c448032 · c448032
1 parent 59ec959
commit c448032
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/paddle/optimizer/optimizer.py b/python/paddle/optimizer/optimizer.py
@@ -1112,8 +1112,7 @@ def clear_grad(self, set_to_zero=True):
 
         if _in_eager_without_dygraph_check():
             for p in param_list:
-                clear_func = p._zero_grads if set_to_zero else p.clear_gradient
-                clear_func()
+                p.clear_gradient(set_to_zero)
         else:
             core.clear_gradients(param_list, set_to_zero)