fix audio shape bug for audio len

4 years ago · 9c0b6c5bb0
parent 718bd30765
commit 9c0b6c5bb0
1 changed files with 3 additions and 3 deletions
--- a/deepspeech/io/collator.py
+++ b/deepspeech/io/collator.py
@ -271,7 +271,7 @@ class SpeechCollator():
            utts.append(utt)
            # audio
            audios.append(audio)  # [T, D]
-            audio_lens.append(audio.shape[1])
+            audio_lens.append(audio.shape[0])
            # text
            # for training, text is token ids
            # else text is string, convert to unicode ord