From 41030f29ebbef862c66b8ab9aad57251c4aadad3 Mon Sep 17 00:00:00 2001
From: sandyhouse <lilong12@baidu.com>
Date: Sun, 7 Mar 2021 20:40:23 +0800
Subject: [PATCH] update

---
 python/paddle/fluid/optimizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
index 6e53f25afc7..633c8f3ab71 100644
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -4917,7 +4917,7 @@ class PipelineOptimizer(object):
                             index=index + 1,
                             type='sum',
                             inputs={'X': [grad_var, param_grad_var]},
-                            outputs={'Out': real_grad_var},
+                            outputs={'Out': param_grad_var},
                             attrs={
                                 #self._op_device_key: device,
                                 self._op_role_key: self._op_role.Backward,
@@ -4954,7 +4954,7 @@ class PipelineOptimizer(object):
                             index=index + 2,
                             type='sum',
                             inputs={'X': [param_grad_var, cast_grad_var]},
-                            outputs={'Out': fp32_grad_var},
+                            outputs={'Out': param_grad_var},
                             attrs={
                                 # self._op_device_key: device,
                                 self._op_role_key: self._op_role.Backward,
-- 
GitLab