Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 41030f29 编写于 作者: S sandyhouse
浏览文件
操作

update

上级 c7f4af1d
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
python/paddle/fluid/optimizer.py
浏览文件 @ 41030f29
...@@ -4917,7 +4917,7 @@ class PipelineOptimizer(object): ...@@ -4917,7 +4917,7 @@ class PipelineOptimizer(object):
index=index + 1, index=index + 1,
type='sum', type='sum',
inputs={'X': [grad_var, param_grad_var]}, inputs={'X': [grad_var, param_grad_var]},
outputs={'Out': real_grad_var}, outputs={'Out': param_grad_var},
attrs={ attrs={
#self._op_device_key: device, #self._op_device_key: device,
self._op_role_key: self._op_role.Backward, self._op_role_key: self._op_role.Backward,
...@@ -4954,7 +4954,7 @@ class PipelineOptimizer(object): ...@@ -4954,7 +4954,7 @@ class PipelineOptimizer(object):
index=index + 2, index=index + 2,
type='sum', type='sum',
inputs={'X': [param_grad_var, cast_grad_var]}, inputs={'X': [param_grad_var, cast_grad_var]},
outputs={'Out': fp32_grad_var}, outputs={'Out': param_grad_var},
attrs={ attrs={
# self._op_device_key: device, # self._op_device_key: device,
self._op_role_key: self._op_role.Backward, self._op_role_key: self._op_role.Backward,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或