🚀 update dependencies and fix rnn layer name

nglehuy · nglehuy · commit cb084901d899 · 2020-12-27T14:15:04.000+07:00
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="nlhuy.cs.16@gmail.com",
     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

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ def __init__(self,`
`45`	`45`	`RNN = get_rnn(rnn_type)`
`46`	`46`	`self.rnn = RNN(`
`47`	`47`	`units=rnn_units, return_sequences=True,`
`48`		`- name=f"{self.name}_rnn", return_state=True,`
	`48`	`+ name=f"{self.name}_{rnn_type}", return_state=True,`
`49`	`49`	`kernel_regularizer=kernel_regularizer,`
`50`	`50`	`bias_regularizer=bias_regularizer`
`51`	`51`	`)`