Merge pull request #154 from TensorSpeech/fix/rnnt

Fix duplicate log softmax
TensorSpeech · Feb 24, 2021 · 5fbd6a8 · 5fbd6a8
2 parents 112a0c3 + bb1584a
commit 5fbd6a8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/setup.py b/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",

diff --git a/tensorflow_asr/losses/rnnt_losses.py b/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")