[ctc] Update search.py (#2398)

According to: https://github.com/wenet-e2e/wenet/blob/a93af33afb29585d7398265f1076d6dec1f11ab8/runtime/core/decoder/ctc_prefix_beam_search.cc#L142
wenet-e2e · Mar 11, 2024 · 68991b4 · 68991b4
1 parent a93af33
commit 68991b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/transformer/search.py b/wenet/transformer/search.py
@@ -175,7 +175,7 @@ def ctc_prefix_beam_search(
                         next_score1.ns = log_add(next_score1.ns,
                                                  prefix_score.ns + prob)
                         if next_score1.v_ns < prefix_score.v_ns + prob:
-                            next_score1.vs_ns = prefix_score.v_ns + prob
+                            next_score1.v_ns = prefix_score.v_ns + prob
                             if next_score1.cur_token_prob < prob:
                                 next_score1.cur_token_prob = prob
                                 next_score1.times_ns = prefix_score.times_ns.copy(