fix initial states (#1398)

Co-authored-by: liujiawang02 <liujiawang02@baidu.com>
k2-fsa · Dec 4, 2023 · f08af2f · f08af2f
1 parent 0622dea
commit f08af2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/decode_stream.py b/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/decode_stream.py
@@ -82,12 +82,12 @@ def __init__(
         self.pad_length = 7
 
         if params.decoding_method == "greedy_search":
-            self.hyp = [params.blank_id] * params.context_size
+            self.hyp = [-1] * (params.context_size - 1) + [params.blank_id]
         elif params.decoding_method == "modified_beam_search":
             self.hyps = HypothesisList()
             self.hyps.add(
                 Hypothesis(
-                    ys=[params.blank_id] * params.context_size,
+                    ys=[-1] * (params.context_size - 1) + [params.blank_id],
                     log_prob=torch.zeros(1, dtype=torch.float32, device=device),
                 )
             )