add modified beam search (#69)

2023-03-01 15:32:54 +08:00
parent e0b76655c8
commit 5326d0f81f
19 changed files with 614 additions and 87 deletions
--- a/sherpa-onnx/csrc/online-transducer-decoder.h
+++ b/sherpa-onnx/csrc/online-transducer-decoder.h
@@ -8,6 +8,7 @@
 #include <vector>

 #include "onnxruntime_cxx_api.h"  // NOLINT
+#include "sherpa-onnx/csrc/hypothesis.h"

 namespace sherpa_onnx {

@@ -17,6 +18,9 @@ struct OnlineTransducerDecoderResult {

  /// number of trailing blank frames decoded so far
  int32_t num_trailing_blanks = 0;
+
+  // used only in modified beam_search
+  Hypotheses hyps;
 };

 class OnlineTransducerDecoder {