🚀 Fix interval functions minor errors

Author: nglehuy

examples/conformer/config.yml

@@ -69,12 +69,12 @@ learning_config:
 
   dataset_config:
     train_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/train-clean-100/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/train-clean-100/transcripts.tsv
     eval_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-clean/transcripts.tsv
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-other/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-other/transcripts.tsv
     test_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/test-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/test-clean/transcripts.tsv
     tfrecords_dir: null
 
   optimizer_config:
@@ -87,7 +87,7 @@ learning_config:
     batch_size: 4
     accumulation_steps: 4
     num_epochs: 20
-    outdir: /mnt/d/SpeechProcessing/Trained/local/conformer
+    outdir: /mnt/d/Models/local/conformer
     log_interval_steps: 300
     eval_interval_steps: 500
     save_interval_steps: 1000

examples/deepspeech2/config.yml

@@ -53,12 +53,12 @@ learning_config:
 
   dataset_config:
     train_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/train-clean-100/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/train-clean-100/transcripts.tsv
     eval_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-clean/transcripts.tsv
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-other/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-other/transcripts.tsv
     test_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/test-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/test-clean/transcripts.tsv
     tfrecords_dir: null
 
   optimizer_config:
@@ -70,7 +70,7 @@ learning_config:
     batch_size: 4
     num_epochs: 20
     accumulation_steps: 8
-    outdir: /mnt/d/SpeechProcessing/Trained/local/deepspeech2
+    outdir: /mnt/d/Models/local/deepspeech2
     log_interval_steps: 400
     save_interval_steps: 400
     eval_interval_steps: 800

examples/jasper/config.yml

@@ -60,12 +60,12 @@ learning_config:
 
   dataset_config:
     train_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/train-clean-100/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/train-clean-100/transcripts.tsv
     eval_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-clean/transcripts.tsv
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-other/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-other/transcripts.tsv
     test_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/test-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/test-clean/transcripts.tsv
     tfrecords_dir: null
 
   optimizer_config:
@@ -77,7 +77,7 @@ learning_config:
     batch_size: 4
     num_epochs: 20
     accumulation_steps: 8
-    outdir: /mnt/d/SpeechProcessing/Trained/local/jasper
+    outdir: /mnt/d/Models/local/jasper
     log_interval_steps: 400
     save_interval_steps: 400
     eval_interval_steps: 800

examples/streaming_transducer/config.yml

@@ -63,12 +63,12 @@ learning_config:
 
   dataset_config:
     train_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/train-clean-100/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/train-clean-100/transcripts.tsv
     eval_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-clean/transcripts.tsv
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/dev-other/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/dev-other/transcripts.tsv
     test_paths:
-      - /mnt/d/SpeechProcessing/Datasets/LibriSpeech/test-clean/transcripts.tsv
+      - /mnt/d/Datasets/Speech/LibriSpeech/test-clean/transcripts.tsv
     tfrecords_dir: null
 
   optimizer_config:
@@ -80,7 +80,7 @@ learning_config:
     batch_size: 2
     accumulation_steps: 1
     num_epochs: 20
-    outdir: /mnt/d/SpeechProcessing/Trained/local/streaming_transducer
+    outdir: /mnt/d/Models/local/streaming_transducer
     log_interval_steps: 300
     eval_interval_steps: 500
     save_interval_steps: 1000

setup.py

@@ -37,7 +37,7 @@
 
 setuptools.setup(
     name="TensorFlowASR",
-    version="0.3.1",
+    version="0.3.2",
     author="Huy Le Nguyen",
     author_email="[email protected]",
     description="Almost State-of-the-art Automatic Speech Recognition using Tensorflow 2",

tensorflow_asr/runners/base_runners.py

@@ -315,23 +315,24 @@ def fit(self, train_dataset, eval_dataset=None, train_bs=None, train_acs=None, e
 
     def _check_log_interval(self):
         """Save log interval."""
-        if self.steps % self.config.log_interval_steps == 0 or \
-                self.steps >= self.total_train_steps:
+        if (self.steps % self.config.log_interval_steps == 0) or \
+                (self.total_train_steps and self.steps >= self.total_train_steps):
             self._write_to_tensorboard(self.train_metrics, self.steps, stage="train")
             """Reset train metrics after save it to tensorboard."""
             for metric in self.train_metrics.keys():
                 self.train_metrics[metric].reset_states()
 
     def _check_save_interval(self):
         """Save log interval."""
-        if self.steps % self.config.save_interval_steps == 0 or \
-                self.steps >= self.total_train_steps:
+        if (self.steps % self.config.save_interval_steps == 0) or \
+                (self.total_train_steps and self.steps >= self.total_train_steps):
             self.save_checkpoint()
             self.save_model_weights()
 
     def _check_eval_interval(self):
         """Save log interval."""
-        if self.steps % self.config.eval_interval_steps == 0:
+        if (self.steps % self.config.eval_interval_steps == 0) or \
+                (self.total_train_steps and self.steps >= self.total_train_steps):
             self._eval_epoch()
 
     # -------------------------------- UTILS -------------------------------------