EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Fangjun Kuang
2024-05-28 17:06:17 +08:00
committed by GitHub
parent 5860e45b4c
commit 50a2eaa41f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sherpa-onnx/csrc/online-recognizer-transducer-impl.h
View File

@@ -344,6 +344,9 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl {
} }
} }
// reset encoder states
s->SetStates(model_->GetEncoderInitStates());
// we keep the decoder_out // we keep the decoder_out
decoder_->UpdateDecoderOut(&s->GetResult()); decoder_->UpdateDecoderOut(&s->GetResult());
Ort::Value decoder_out = std::move(s->GetResult().decoder_out); Ort::Value decoder_out = std::move(s->GetResult().decoder_out);