Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
DeepSpeech
提交
9560d650
D
DeepSpeech
项目概览
PaddlePaddle
/
DeepSpeech
大约 2 年 前同步成功
通知
210
Star
8425
Fork
1598
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
245
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
DeepSpeech
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
245
Issue
245
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
9560d650
编写于
9月 09, 2022
作者:
T
tianhao zhang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix dp init
上级
82e04d78
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
1 addition
and
10 deletion
+1
-10
paddlespeech/s2t/training/trainer.py
paddlespeech/s2t/training/trainer.py
+1
-10
未找到文件。
paddlespeech/s2t/training/trainer.py
浏览文件 @
9560d650
...
@@ -19,9 +19,8 @@ from pathlib import Path
...
@@ -19,9 +19,8 @@ from pathlib import Path
import
paddle
import
paddle
from
paddle
import
distributed
as
dist
from
paddle
import
distributed
as
dist
import
pdb
pdb
.
set_trace
()
dist
.
init_parallel_env
()
dist
.
init_parallel_env
()
from
visualdl
import
LogWriter
from
visualdl
import
LogWriter
from
paddlespeech.s2t.training.reporter
import
ObsScope
from
paddlespeech.s2t.training.reporter
import
ObsScope
...
@@ -125,9 +124,6 @@ class Trainer():
...
@@ -125,9 +124,6 @@ class Trainer():
else
:
else
:
raise
Exception
(
"invalid device"
)
raise
Exception
(
"invalid device"
)
if
self
.
parallel
:
self
.
init_parallel
()
self
.
checkpoint
=
Checkpoint
(
self
.
checkpoint
=
Checkpoint
(
kbest_n
=
self
.
config
.
checkpoint
.
kbest_n
,
kbest_n
=
self
.
config
.
checkpoint
.
kbest_n
,
latest_n
=
self
.
config
.
checkpoint
.
latest_n
)
latest_n
=
self
.
config
.
checkpoint
.
latest_n
)
...
@@ -176,11 +172,6 @@ class Trainer():
...
@@ -176,11 +172,6 @@ class Trainer():
"""
"""
return
self
.
args
.
ngpu
>
1
return
self
.
args
.
ngpu
>
1
def
init_parallel
(
self
):
"""Init environment for multiprocess training.
"""
# dist.init_parallel_env()
@
mp_tools
.
rank_zero_only
@
mp_tools
.
rank_zero_only
def
save
(
self
,
tag
=
None
,
infos
:
dict
=
None
):
def
save
(
self
,
tag
=
None
,
infos
:
dict
=
None
):
"""Save checkpoint (model parameters and optimizer states).
"""Save checkpoint (model parameters and optimizer states).
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录