Fix LogAdd (#316)

Using 0 as the initial value, should not perform addition when both values are 0
2023-09-18 10:43:04 +08:00
parent c471423125
commit 5ca0ff8811
1 changed files with 3 additions and 1 deletions
--- a/sherpa-onnx/csrc/hypothesis.cc
+++ b/sherpa-onnx/csrc/hypothesis.cc
@@ -18,8 +18,10 @@ void Hypotheses::Add(Hypothesis hyp) {
  } else {
    it->second.log_prob = LogAdd<double>()(it->second.log_prob, hyp.log_prob);

-    it->second.lm_log_prob =
+    if (it->second.lm_log_prob != 0 && hyp.lm_log_prob != 0){
+      it->second.lm_log_prob =
        LogAdd<double>()(it->second.lm_log_prob, hyp.lm_log_prob);
+    }
  }
 }