🚀 Fixed tfrecords and summary

Author: nglehuy

setup.py

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

tensorflow_asr/datasets/asr_dataset.py

@@ -164,16 +164,16 @@ def create_tfrecords(self):
         if not os.path.exists(self.tfrecords_dir):
             os.makedirs(self.tfrecords_dir)
 
-        entries = self.read_entries()
-        if len(entries) <= 0:
-            return False
-
         if glob.glob(os.path.join(self.tfrecords_dir, f"{self.stage}*.tfrecord")):
             print(f"TFRecords're already existed: {self.stage}")
             return True
 
         print(f"Creating {self.stage}.tfrecord ...")
 
+        entries = self.read_entries()
+        if len(entries) <= 0:
+            return False
+
         def get_shard_path(shard_id):
             return os.path.join(self.tfrecords_dir, f"{self.stage}_{shard_id}.tfrecord")
 

tensorflow_asr/models/transducer.py

@@ -242,7 +242,6 @@ def _build(self, input_shape):
         self([inputs, input_length, pred, pred_length], training=False)
 
     def summary(self, line_length=None, **kwargs):
-        self.encoder.summary(line_length=line_length, **kwargs)
         self.predict_net.summary(line_length=line_length, **kwargs)
         self.joint_net.summary(line_length=line_length, **kwargs)
         super(Transducer, self).summary(line_length=line_length, **kwargs)