Reset encoder states on endpointing for streaming transducer. (#924)

k2-fsa · May 28, 2024 · 50a2eaa · 50a2eaa
1 parent 5860e45
commit 50a2eaa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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);