Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
2befb9f9
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
2befb9f9
编写于
10月 14, 2017
作者:
Q
qiaolongfei
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
optimizer backward CreateGradVarInBlock input output order
上级
9a0ef7d2
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
7 addition
and
7 deletion
+7
-7
paddle/framework/backward.cc
paddle/framework/backward.cc
+7
-6
python/paddle/v2/framework/tests/test_program.py
python/paddle/v2/framework/tests/test_program.py
+0
-1
未找到文件。
paddle/framework/backward.cc
浏览文件 @
2befb9f9
...
...
@@ -274,9 +274,10 @@ static bool AllGradInSet(const std::vector<std::string>& names,
}
static
void
CreateGradVarInBlock
(
std
::
unordered_map
<
std
::
string
,
GradVarInfo
>*
grad_var_record
,
BlockDescBind
*
block_desc
,
size_t
grad_op_start_index
,
const
std
::
unordered_map
<
std
::
string
,
std
::
string
>&
param_name_map
)
{
size_t
grad_op_start_index
,
const
std
::
unordered_map
<
std
::
string
,
std
::
string
>&
param_name_map
,
BlockDescBind
*
block_desc
,
std
::
unordered_map
<
std
::
string
,
GradVarInfo
>*
grad_var_record
)
{
auto
ops
=
block_desc
->
AllOps
();
for
(
size_t
op_index
=
grad_op_start_index
;
op_index
<
ops
.
size
();
++
op_index
)
{
...
...
@@ -451,11 +452,11 @@ ParamGradInfoMap AppendBackward(
root_block
->
NewVar
(
fill_one_op_out
);
// create grad_var for all blocks in this program
CreateGradVarInBlock
(
&
retv
,
root_block
,
forward_op_num
,
grad_to_var
);
CreateGradVarInBlock
(
forward_op_num
,
grad_to_var
,
root_block
,
&
retv
);
for
(
size_t
block_index
=
forward_block_num
;
block_index
<
program_desc
.
Size
();
++
block_index
)
{
CreateGradVarInBlock
(
&
retv
,
program_desc
.
Block
(
block_index
),
0
,
grad_to_var
);
CreateGradVarInBlock
(
0
,
grad_to_var
,
program_desc
.
Block
(
block_index
)
,
&
retv
);
}
return
retv
;
}
...
...
python/paddle/v2/framework/tests/test_program.py
浏览文件 @
2befb9f9
...
...
@@ -69,7 +69,6 @@ class TestProgram(unittest.TestCase):
for
op
in
block
.
all_ops
():
actual_ops
.
append
(
op
.
type
())
print
(
actual_ops
)
self
.
assertEqual
(
actual_ops
,
expect_ops
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录