From 3d5c7f3359ac68d24d22979a6f00fc2022515663 Mon Sep 17 00:00:00 2001 From: Huy Le Nguyen Date: Wed, 6 Jan 2021 00:56:53 +0700 Subject: [PATCH] :writing_hand: fix transducer beam search for longer sequence --- tensorflow_asr/models/transducer.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tensorflow_asr/models/transducer.py b/tensorflow_asr/models/transducer.py index 15b12a6c88..3a283c6ad8 100755 --- a/tensorflow_asr/models/transducer.py +++ b/tensorflow_asr/models/transducer.py @@ -591,7 +591,7 @@ def initialize_beam(dynamic=False): B = BeamHypothesis( score=B.score.write(0, 0.0), indices=B.indices.write(0, self.text_featurizer.blank), - prediction=B.prediction.write(0, tf.ones([total], dtype=tf.int32) * self.text_featurizer.blank), + prediction=B.prediction.write(0, tf.ones([total * 2], dtype=tf.int32) * self.text_featurizer.blank), states=B.states.write(0, self.predict_net.get_initial_state()) ) @@ -673,10 +673,7 @@ def false_fn(): b_score, b_indices, b_prediction, b_states, \ a_score, a_indices, a_prediction, a_states, A_i = tf.cond( - tf.equal(pred, self.text_featurizer.blank), - true_fn=true_fn, - false_fn=false_fn - ) + tf.equal(pred, self.text_featurizer.blank), true_fn=true_fn, false_fn=false_fn) B = BeamHypothesis(score=b_score, indices=b_indices, prediction=b_prediction, states=b_states) A = BeamHypothesis(score=a_score, indices=a_indices, prediction=a_prediction, states=a_states)