mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Using non strict match in context graph for contextual words (#1952)

Browse Source
This commit is contained in:
Wei Kang 2025-06-19 12:27:15 +08:00 committed by GitHub
parent f80a2ee110
commit 343b8fa2dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py
View File

@ -1347,7 +1347,10 @@ def modified_beam_search(
(
context_score,
new_context_state,
) = context_graph.forward_one_step(hyp.context_state, new_token)
_,
) = context_graph.forward_one_step(
hyp.context_state, new_token, strict_mode=False
)
new_log_prob = topk_log_probs[k] + context_score
@ -2853,7 +2856,10 @@ def modified_beam_search_LODR(
(
context_score,
new_context_state,
) = context_graph.forward_one_step(hyp.context_state, new_token)
_,
) = context_graph.forward_one_step(
hyp.context_state, new_token, strict_mode=False
)
ys.append(new_token)
state_cost = hyp.state_cost.forward_one_step(new_token)