Merge pull request #70 from TensorSpeech/fix/transducer

Fix typo and format for transducer

Author: nglehuy

.gitignore

@@ -9,4 +9,5 @@ Session.vim
 .idea
 .vscode
 __pycache__
+.pytest*
 venv

setup.cfg

@@ -1,8 +1,8 @@
 [flake8]
 ignore = E402,E701,E702,E704,E251
-max-line-length = 150
+max-line-length = 127
 
 [pep8]
 ignore = E402,E701,E702,E704,E251
-max-line-length = 150
+max-line-length = 127
 indent-size = 4

tensorflow_asr/models/streaming_transducer.py

@@ -257,7 +257,7 @@ def recognize(self, signals):
         """
         def execute(signal: tf.Tensor):
             features = self.speech_featurizer.tf_extract(signal)
-            encoded, _ = self.encoder_inference(features, self.encoder.get_initial_states())
+            encoded, _ = self.encoder_inference(features, self.encoder.get_initial_state())
             hypothesis = self.perform_greedy(
                 encoded,
                 predicted=tf.constant(self.text_featurizer.blank, dtype=tf.int32),
@@ -310,10 +310,13 @@ def recognize_beam(self, signals, lm=False):
         """
         def execute(signal: tf.Tensor):
             features = self.speech_featurizer.tf_extract(signal)
-            encoded, _ = self.encoder_inference(features, self.encoder.get_initial_states())
+            encoded, _ = self.encoder_inference(features, self.encoder.get_initial_state())
             hypothesis = self.perform_beam_search(encoded, lm)
-            prediction = tf.map_fn(lambda x: tf.strings.to_number(x, tf.int32),
-                                   tf.strings.split(hypothesis.prediction), fn_output_signature=tf.TensorSpec([], dtype=tf.int32))
+            prediction = tf.map_fn(
+                lambda x: tf.strings.to_number(x, tf.int32),
+                tf.strings.split(hypothesis.prediction),
+                fn_output_signature=tf.TensorSpec([], dtype=tf.int32)
+            )
             transcripts = self.text_featurizer.iextract(tf.expand_dims(prediction, axis=0))
             return tf.squeeze(transcripts)  # reshape from [1] to []
 

tensorflow_asr/models/transducer.py

@@ -451,8 +451,11 @@ def execute(signal: tf.Tensor):
             features = self.speech_featurizer.tf_extract(signal)
             encoded = self.encoder_inference(features)
             hypothesis = self.perform_beam_search(encoded, lm)
-            prediction = tf.map_fn(lambda x: tf.strings.to_number(x, tf.int32),
-                                   tf.strings.split(hypothesis.prediction), fn_output_signature=tf.TensorSpec([], dtype=tf.int32))
+            prediction = tf.map_fn(
+                lambda x: tf.strings.to_number(x, tf.int32),
+                tf.strings.split(hypothesis.prediction),
+                fn_output_signature=tf.TensorSpec([], dtype=tf.int32)
+            )
             transcripts = self.text_featurizer.iextract(tf.expand_dims(prediction, axis=0))
             return tf.squeeze(transcripts)  # reshape from [1] to []
 

tests/plot_learning_rate.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import math
 import tensorflow as tf
 import matplotlib.pyplot as plt
 from tensorflow_asr.optimizers.schedules import SANSchedule, TransformerSchedule