Merge pull request #154 from TensorSpeech/fix/rnnt

Fix duplicate log softmax

Author: nglehuy

setup.py

@@ -22,7 +22,7 @@
 
 setuptools.setup(
     name="TensorFlowASR",
-    version="0.7.7",
+    version="0.7.8",
     author="Huy Le Nguyen",
     author_email="[email protected]",
     description="Almost State-of-the-art Automatic Speech Recognition using Tensorflow 2",

tensorflow_asr/losses/rnnt_losses.py

@@ -235,7 +235,6 @@ def rnnt_loss_tf(logits, labels, label_length, logit_length, name=None):
     name = "rnnt_loss" if name is None else name
     with tf.name_scope(name):
         logits = tf.convert_to_tensor(logits, name="logits")
-        logits = tf.nn.log_softmax(logits)
         labels = tf.convert_to_tensor(labels, name="labels")
         label_length = tf.convert_to_tensor(label_length, name="label_length")
         logit_length = tf.convert_to_tensor(logit_length, name="logit_length")