Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
3901b2cf
P
Paddle
项目概览
Crayon鑫
/
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看板
未验证
提交
3901b2cf
编写于
6月 22, 2022
作者:
X
xiongkun
提交者:
GitHub
6月 22, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix dygraph2static bugs (#43688)
因此将N次前向共用一个 vec scope 换为了N次前向分别使用自己的一个vecscope
上级
fc5a85b0
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
1 addition
and
10 deletion
+1
-10
python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py
...paddle/fluid/dygraph/dygraph_to_static/partial_program.py
+1
-10
未找到文件。
python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py
浏览文件 @
3901b2cf
...
...
@@ -152,7 +152,6 @@ class PartialProgramLayer:
assert
isinstance
(
self
.
_build_strategy
,
BuildStrategy
)
self
.
_origin_main_program
=
self
.
_verify_program
(
main_program
)
self
.
_tmp_scope_vec
=
self
.
_create_scope_vec
()
self
.
_cuda_graph_vec
=
self
.
_create_cuda_graph_vec
()
self
.
_cuda_graph_capture_mode
=
""
self
.
_cuda_graph_pool_id
=
0
...
...
@@ -363,9 +362,8 @@ class PartialProgramLayer:
_C_ops
.
run_program
(
self
.
_valid_vars
(
in_vars
),
self
.
_valid_vars
(
self
.
_params
),
self
.
_valid_vars
(
out_vars
),
self
.
_
tmp_scope_vec
,
self
.
_valid_vars
(
out_vars
),
self
.
_
create_scope_vec
()
,
self
.
_double_grads
,
self
.
_cuda_graph_vec
,
*
attrs
)
self
.
drop_scope_if_no_grad
()
restored_nest_out
=
self
.
_restore_out
(
out_vars
)
return
self
.
_remove_no_value
(
restored_nest_out
)
...
...
@@ -379,13 +377,6 @@ class PartialProgramLayer:
in_vars
[
i
]
=
var
.
astype
(
'float16'
)
in_vars
[
i
].
name
=
name
def
drop_scope_if_no_grad
(
self
):
tracer
=
framework
.
_dygraph_tracer
()
scope
=
self
.
_tmp_scope_vec
.
value
().
get_scope
()
if
isinstance
(
self
.
_tmp_scope_vec
,
(
core
.
VarBase
))
else
self
.
_tmp_scope_vec
[
0
]
if
self
.
training
and
not
tracer
.
_has_grad
:
scope
.
drop_kids
()
@
property
def
program
(
self
):
if
self
.
training
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录