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

Update offline-ctc-greedy-search-decoder.cc (#917)

Browse Source 
Bug fixes.
Z_O_O will be decoded as ZO instead of ZOO.
To fix this, prev_id should update every time.
This commit is contained in:
Dadoou
2024-05-24 22:31:56 +08:00
committed by GitHub
parent 86ccb86006
commit 4fc0a1dc64
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sherpa-onnx/csrc/offline-ctc-greedy-search-decoder.cc
View File

@@ -42,8 +42,8 @@ std::vector<OfflineCtcDecoderResult> OfflineCtcGreedySearchDecoder::Decode(
if (y != blank_id_ && y != prev_id) { if (y != blank_id_ && y != prev_id) {
r.tokens.push_back(y); r.tokens.push_back(y);
r.timestamps.push_back(t); r.timestamps.push_back(t);
prev_id = y;
} }
prev_id = y;
} // for (int32_t t = 0; ...) } // for (int32_t t = 0; ...)
ans.push_back(std::move(r)); ans.push_back(std::move(r));