Merge pull request TensorSpeech#113 from TensorSpeech/fix/beamsearch

Fix padding in transducer batch beam search
karibbov · Jan 14, 2021 · f23235e · f23235e
2 parents 8b9383c + c4ab7dc
commit f23235e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_asr/models/transducer.py b/tensorflow_asr/models/transducer.py
@@ -538,7 +538,7 @@ def body(batch, decoded):
             )
             prediction = tf.pad(
                 hypothesis.prediction,
-                paddings=[[0, total_time - encoded_length[batch]]],
+                paddings=[[0, 2 * (total_time - encoded_length[batch])]],
                 mode="CONSTANT", constant_values=self.text_featurizer.blank
             )
             decoded = decoded.write(batch, prediction)