with @qiantong fix lexicon decoder for --surround usage with CTC crit…

…erion
flashlight · Oct 16, 2020 · b1d1f89 · b1d1f89
1 parent 96f5f9d
commit b1d1f89
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Decode.cpp b/Decode.cpp
@@ -251,7 +251,7 @@ int main(int argc, char** argv) {
       FLAGS_criterion == kCtcCriterion ? tokenDict.getIndex(kBlankToken) : -1;
   int silIdx = -1;
   if (FLAGS_wordseparator != "") {
-    tokenDict.getIndex(FLAGS_wordseparator);
+    silIdx = tokenDict.getIndex(FLAGS_wordseparator);
   }
   std::shared_ptr<Trie> trie = nullptr;
   if (FLAGS_decodertype == "wrd" || FLAGS_uselexicon) {

diff --git a/src/libraries/decoder/LexiconDecoder.cpp b/src/libraries/decoder/LexiconDecoder.cpp
@@ -154,8 +154,9 @@ void LexiconDecoder::decodeStep(const float* emissions, int T, int N) {
       }
 
       /* (2) Try same lexicon node */
-      if (opt_.criterionType != CriterionType::CTC || !prevHyp.prevBlank) {
-        int n = prevIdx;
+      if (opt_.criterionType != CriterionType::CTC || !prevHyp.prevBlank ||
+          prevLex == lexicon_->getRoot()) {
+        int n = prevLex == lexicon_->getRoot() ? sil_ : prevIdx;
         double amScore = emissions[t * N + n];
         if (nDecodedFrames_ + t > 0 &&
             opt_.criterionType == CriterionType::ASG) {