Update am_dataloader.py

phecda-xu · Jan 3, 2021 · 926832c · 926832c
1 parent da46bcc
commit 926832c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataloaders/am_dataloader.py b/dataloaders/am_dataloader.py
@@ -39,7 +39,7 @@ def return_data_types(self):
         if self.LAS:
             return (tf.float32, tf.float32, tf.int32, tf.int32, tf.int32,tf.float32)
         else:
-            return  (tf.float32, tf.float32, tf.int32, tf.int32, tf.int32)
+            return  (tf.float32, tf.int32, tf.int32, tf.int32)
     def return_data_shape(self):
         f,c=self.speech_featurizer.compute_feature_dim()
         if self.LAS: