Fix a bug in mean calculation of 'ys_probs' (#748)

k2-fsa · Apr 11, 2024 · 904a3cc · 904a3cc
1 parent d21c45d
commit 904a3cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sherpa-onnx/csrc/transducer-keyword-decoder.cc b/sherpa-onnx/csrc/transducer-keyword-decoder.cc
@@ -152,7 +152,7 @@ void TransducerKeywordDecoder::Decode(
       if (matched) {
         float ys_prob = 0.0;
         int32_t length = best_hyp.ys_probs.size();
-        for (int32_t i = 1; i <= matched_state->level; ++i) {
+        for (int32_t i = 0; i < matched_state->level; ++i) {
           ys_prob += best_hyp.ys_probs[i];
         }
         ys_prob /= matched_state->level;