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

csukuangfj · web-flow · commit 99b65aaef263 · 2024-05-28T17:06:17.000+08:00
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);

Original file line number	Diff line number	Diff line change
`@@ -344,6 +344,9 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl {`
`344`	`344`	`}`
`345`	`345`	`}`
`346`	`346`
	`347`	`+ // reset encoder states`
	`348`	`+ s->SetStates(model_->GetEncoderInitStates());`
	`349`	`+`
`347`	`350`	`// we keep the decoder_out`
`348`	`351`	`decoder_->UpdateDecoderOut(&s->GetResult());`
`349`	`352`	`Ort::Value decoder_out = std::move(s->GetResult().decoder_out);`