Fix for beam alignment

amoliu · Oct 27, 2017 · 2e9addd · 2e9addd
1 parent 52a64a8
commit 2e9addd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch_ctc/src/ctc_beam_search.h b/pytorch_ctc/src/ctc_beam_search.h
@@ -154,7 +154,6 @@ Status CTCBeamSearchDecoder<CTCBeamState, CTCBeamComparer>::Decode(
   // Storage for top paths.
   std::vector<std::vector<int>> beams;
   std::vector<float> beam_log_probabilities;
-  std::vector<std::vector<int>> beam_alignments;
   int top_n = output->size();
   if (std::any_of(output->begin(), output->end(),
                   [batch_size_](const CTCDecoder::Output& output) -> bool {
@@ -188,7 +187,7 @@ Status CTCBeamSearchDecoder<CTCBeamState, CTCBeamComparer>::Decode(
       leaves_.push(entry);
     }
 
-
+    std::vector<std::vector<int>> beam_alignments;
     Status status = TopPaths(top_n, &beams, &beam_log_probabilities, &beam_alignments,
                     merge_repeated_);