Fangjun Kuang
Committed by GitHub

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

@@ -344,6 +344,9 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl { @@ -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 // we keep the decoder_out 350 // we keep the decoder_out
348 decoder_->UpdateDecoderOut(&s->GetResult()); 351 decoder_->UpdateDecoderOut(&s->GetResult());
349 Ort::Value decoder_out = std::move(s->GetResult().decoder_out); 352 Ort::Value decoder_out = std::move(s->GetResult().decoder_out);