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

Using non strict match in context graph for contextual words

Browse Source
This commit is contained in:
pkufool 2025-05-23 17:43:30 +08:00
parent 021e1a8846
commit a6aaf33843
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, context_score,
new_context_state, 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 new_log_prob = topk_log_probs[k] + context_score
@ -2853,7 +2856,10 @@ def modified_beam_search_LODR(
( (
context_score, context_score,
new_context_state, 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) ys.append(new_token)
state_cost = hyp.state_cost.forward_one_step(new_token) state_cost = hyp.state_cost.forward_one_step(new_token)