Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
DeepSpeech
提交
7c884344
D
DeepSpeech
项目概览
PaddlePaddle
/
DeepSpeech
大约 1 年 前同步成功
通知
207
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
7c884344
编写于
11月 09, 2021
作者:
J
Junkun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add word reward into beam search.
上级
e5edc83a
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
7 addition
and
4 deletion
+7
-4
examples/ted_en_zh/t0/conf/transformer_joint_noam.yaml
examples/ted_en_zh/t0/conf/transformer_joint_noam.yaml
+1
-0
paddlespeech/s2t/exps/u2_st/model.py
paddlespeech/s2t/exps/u2_st/model.py
+2
-4
paddlespeech/s2t/models/u2_st/u2_st.py
paddlespeech/s2t/models/u2_st/u2_st.py
+4
-0
未找到文件。
examples/ted_en_zh/t0/conf/transformer_joint_noam.yaml
浏览文件 @
7c884344
...
...
@@ -99,6 +99,7 @@ decoding:
alpha
:
2.5
beta
:
0.3
beam_size
:
10
word_reward
:
0.7
cutoff_prob
:
1.0
cutoff_top_n
:
0
num_proc_bsearch
:
8
...
...
paddlespeech/s2t/exps/u2_st/model.py
浏览文件 @
7c884344
...
...
@@ -441,10 +441,7 @@ class U2STTester(U2STTrainer):
""
.
join
(
chr
(
t
)
for
t
in
text
[:
text_len
])
for
text
,
text_len
in
zip
(
texts
,
texts_len
)
]
# from IPython import embed
# import os
# embed()
# os._exit(0)
hyps
=
self
.
model
.
decode
(
audio
,
audio_len
,
...
...
@@ -458,6 +455,7 @@ class U2STTester(U2STTrainer):
cutoff_top_n
=
cfg
.
cutoff_top_n
,
num_processes
=
cfg
.
num_proc_bsearch
,
ctc_weight
=
cfg
.
ctc_weight
,
word_reward
=
cfg
.
word_reward
,
decoding_chunk_size
=
cfg
.
decoding_chunk_size
,
num_decoding_left_chunks
=
cfg
.
num_decoding_left_chunks
,
simulate_streaming
=
cfg
.
simulate_streaming
)
...
...
paddlespeech/s2t/models/u2_st/u2_st.py
浏览文件 @
7c884344
...
...
@@ -315,6 +315,7 @@ class U2STBaseModel(nn.Layer):
speech
:
paddle
.
Tensor
,
speech_lengths
:
paddle
.
Tensor
,
beam_size
:
int
=
10
,
word_reward
:
float
=
0.0
,
decoding_chunk_size
:
int
=-
1
,
num_decoding_left_chunks
:
int
=-
1
,
simulate_streaming
:
bool
=
False
,
)
->
paddle
.
Tensor
:
...
...
@@ -378,6 +379,7 @@ class U2STBaseModel(nn.Layer):
# 2.2 First beam prune: select topk best prob at current time
top_k_logp
,
top_k_index
=
logp
.
topk
(
beam_size
)
# (B*N, N)
top_k_logp
+=
word_reward
top_k_logp
=
mask_finished_scores
(
top_k_logp
,
end_flag
)
top_k_index
=
mask_finished_preds
(
top_k_index
,
end_flag
,
self
.
eos
)
...
...
@@ -528,6 +530,7 @@ class U2STBaseModel(nn.Layer):
cutoff_top_n
:
int
,
num_processes
:
int
,
ctc_weight
:
float
=
0.0
,
word_reward
:
float
=
0.0
,
decoding_chunk_size
:
int
=-
1
,
num_decoding_left_chunks
:
int
=-
1
,
simulate_streaming
:
bool
=
False
):
...
...
@@ -569,6 +572,7 @@ class U2STBaseModel(nn.Layer):
feats
,
feats_lengths
,
beam_size
=
beam_size
,
word_reward
=
word_reward
,
decoding_chunk_size
=
decoding_chunk_size
,
num_decoding_left_chunks
=
num_decoding_left_chunks
,
simulate_streaming
=
simulate_streaming
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录