Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
4a1bdf58
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
4a1bdf58
编写于
5月 04, 2023
作者:
D
duanyanhui
提交者:
cuicheng01
5月 05, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
rm with_data_parallel
上级
6a55aac3
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
4 addition
and
15 deletion
+4
-15
ppcls/static/program.py
ppcls/static/program.py
+4
-15
未找到文件。
ppcls/static/program.py
浏览文件 @
4a1bdf58
...
...
@@ -153,12 +153,6 @@ def create_strategy(config):
exec_strategy: exec strategy
"""
build_strategy
=
paddle
.
static
.
BuildStrategy
()
exec_strategy
=
paddle
.
static
.
ExecutionStrategy
()
exec_strategy
.
num_threads
=
1
exec_strategy
.
num_iteration_per_drop_scope
=
(
10000
if
'AMP'
in
config
and
config
.
AMP
.
get
(
"level"
,
"O1"
)
==
"O2"
else
10
)
fuse_op
=
True
if
'AMP'
in
config
else
False
...
...
@@ -172,7 +166,7 @@ def create_strategy(config):
build_strategy
.
fuse_bn_add_act_ops
=
fuse_bn_add_act_ops
build_strategy
.
enable_addto
=
enable_addto
return
build_strategy
,
exec_strategy
return
build_strategy
def
dist_optimizer
(
config
,
optimizer
):
...
...
@@ -186,10 +180,9 @@ def dist_optimizer(config, optimizer):
Returns:
optimizer: a distributed optimizer
"""
build_strategy
,
exec_strategy
=
create_strategy
(
config
)
build_strategy
=
create_strategy
(
config
)
dist_strategy
=
DistributedStrategy
()
dist_strategy
.
execution_strategy
=
exec_strategy
dist_strategy
.
build_strategy
=
build_strategy
dist_strategy
.
nccl_comm_num
=
1
...
...
@@ -298,14 +291,10 @@ def compile(config, program, loss_name=None, share_prog=None):
Returns:
compiled_program(): a compiled program
"""
build_strategy
,
exec_strategy
=
create_strategy
(
config
)
build_strategy
=
create_strategy
(
config
)
compiled_program
=
paddle
.
static
.
CompiledProgram
(
program
).
with_data_parallel
(
share_vars_from
=
share_prog
,
loss_name
=
loss_name
,
build_strategy
=
build_strategy
,
exec_strategy
=
exec_strategy
)
program
,
build_strategy
=
build_strategy
)
return
compiled_program
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录