⚡ Update streaming transducer example

Author: nglehuy

examples/streaming_transducer/config.yml

@@ -71,6 +71,11 @@ learning_config:
       - /mnt/Data/ML/ASR/Raw/LibriSpeech/test-clean/transcripts.tsv
     tfrecords_dir: null
 
+  optimizer_config:
+    class_name: adam
+    config:
+      learning_rate: 0.0001
+
   running_config:
     batch_size: 2
     accumulation_steps: 1

examples/streaming_transducer/train_ga_streaming_transducer.py

@@ -111,7 +111,7 @@
     streaming_transducer._build(speech_featurizer.shape)
     streaming_transducer.summary(line_length=150)
 
-    optimizer = tf.keras.optimizers.Adam()
+    optimizer = tf.keras.optimizers.get(config.learning_config.optimizer_config)
 
 streaming_transducer_trainer.compile(model=streaming_transducer, optimizer=optimizer,
                                      max_to_keep=args.max_ckpts)

examples/streaming_transducer/train_ga_subword_streaming_transducer.py

@@ -127,7 +127,7 @@
     streaming_transducer._build(speech_featurizer.shape)
     streaming_transducer.summary(line_length=150)
 
-    optimizer = tf.keras.optimizers.Adam()
+    optimizer = tf.keras.optimizers.get(config.learning_config.optimizer_config)
 
 streaming_transducer_trainer.compile(model=streaming_transducer, optimizer=optimizer,
                                      max_to_keep=args.max_ckpts)

examples/streaming_transducer/train_streaming_transducer.py

@@ -111,7 +111,7 @@
     streaming_transducer._build(speech_featurizer.shape)
     streaming_transducer.summary(line_length=150)
 
-    optimizer = tf.keras.optimizers.Adam()
+    optimizer = tf.keras.optimizers.get(config.learning_config.optimizer_config)
 
 streaming_transducer_trainer.compile(model=streaming_transducer, optimizer=optimizer,
                                      max_to_keep=args.max_ckpts)

examples/streaming_transducer/train_subword_streaming_transducer.py

@@ -127,7 +127,7 @@
     streaming_transducer._build(speech_featurizer.shape)
     streaming_transducer.summary(line_length=150)
 
-    optimizer = tf.keras.optimizers.Adam()
+    optimizer = tf.keras.optimizers.get(config.learning_config.optimizer_config)
 
 streaming_transducer_trainer.compile(model=streaming_transducer, optimizer=optimizer,
                                      max_to_keep=args.max_ckpts)