From 50a2eaa41f142c5deabc90ea8417aecca1581d11 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Tue, 28 May 2024 17:06:17 +0800 Subject: [PATCH] Reset encoder states on endpointing for streaming transducer. (#924) --- sherpa-onnx/csrc/online-recognizer-transducer-impl.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sherpa-onnx/csrc/online-recognizer-transducer-impl.h b/sherpa-onnx/csrc/online-recognizer-transducer-impl.h index 402346fa8..e7c8fa7e5 100644 --- a/sherpa-onnx/csrc/online-recognizer-transducer-impl.h +++ b/sherpa-onnx/csrc/online-recognizer-transducer-impl.h @@ -344,6 +344,9 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl { } } + // reset encoder states + s->SetStates(model_->GetEncoderInitStates()); + // we keep the decoder_out decoder_->UpdateDecoderOut(&s->GetResult()); Ort::Value decoder_out = std::move(s->GetResult().decoder_out);