🚀 update dependencies and fix rnn layer name

TensorSpeech · Dec 27, 2020 · cb08490 · cb08490
1 parent b771734
commit cb08490
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -29,11 +29,12 @@
     "tqdm>=4.54.1",
     "colorama>=0.4.4",
     "nlpaug>=1.1.1",
+    "nltk>=3.5"
 ]
 
 setuptools.setup(
     name="TensorFlowASR",
-    version="0.6.1",
+    version="0.6.2",
     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/models/streaming_transducer.py b/tensorflow_asr/models/streaming_transducer.py
@@ -45,7 +45,7 @@ def __init__(self,
         RNN = get_rnn(rnn_type)
         self.rnn = RNN(
             units=rnn_units, return_sequences=True,
-            name=f"{self.name}_rnn", return_state=True,
+            name=f"{self.name}_{rnn_type}", return_state=True,
             kernel_regularizer=kernel_regularizer,
             bias_regularizer=bias_regularizer
         )

diff --git a/tensorflow_asr/models/transducer.py b/tensorflow_asr/models/transducer.py
@@ -53,7 +53,7 @@ def __init__(self,
         for i in range(num_rnns):
             rnn = RNN(
                 units=rnn_units, return_sequences=True,
-                name=f"{name}_lstm_{i}", return_state=True,
+                name=f"{name}_{rnn_type}_{i}", return_state=True,
                 implementation=rnn_implementation,
                 kernel_regularizer=kernel_regularizer,
                 bias_regularizer=bias_regularizer