Fix nemo streaming transducer greedy search (#944)

This commit is contained in:
Fangjun Kuang
2024-05-30 15:31:10 +08:00
committed by GitHub
parent 3f472a9993
commit 082f230dfb
18 changed files with 318 additions and 288 deletions

View File

@@ -7,27 +7,22 @@
#define SHERPA_ONNX_CSRC_ONLINE_TRANSDUCER_GREEDY_SEARCH_NEMO_DECODER_H_
#include <vector>
#include "sherpa-onnx/csrc/online-transducer-decoder.h"
#include "sherpa-onnx/csrc/online-transducer-nemo-model.h"
namespace sherpa_onnx {
class OnlineStream;
class OnlineTransducerGreedySearchNeMoDecoder {
public:
OnlineTransducerGreedySearchNeMoDecoder(OnlineTransducerNeMoModel *model,
float blank_penalty)
: model_(model),
blank_penalty_(blank_penalty) {}
: model_(model), blank_penalty_(blank_penalty) {}
OnlineTransducerDecoderResult GetEmptyResult() const;
void UpdateDecoderOut(OnlineTransducerDecoderResult *result) {}
void StripLeadingBlanks(OnlineTransducerDecoderResult * /*r*/) const {}
std::vector<Ort::Value> Decode(
Ort::Value encoder_out,
std::vector<Ort::Value> decoder_states,
std::vector<OnlineTransducerDecoderResult> *result,
OnlineStream **ss = nullptr, int32_t n = 0);
// @param n number of elements in ss
void Decode(Ort::Value encoder_out, OnlineStream **ss, int32_t n) const;
private:
OnlineTransducerNeMoModel *model_; // Not owned
@@ -37,4 +32,3 @@ class OnlineTransducerGreedySearchNeMoDecoder {
} // namespace sherpa_onnx
#endif // SHERPA_ONNX_CSRC_ONLINE_TRANSDUCER_GREEDY_SEARCH_NEMO_DECODER_H_