Committed by
GitHub
Reset encoder states on endpointing for streaming transducer. (#924)
正在显示
1 个修改的文件
包含
3 行增加
和
0 行删除
| @@ -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); |
-
请 注册 或 登录 后发表评论