Merge pull request #691 from PaddlePaddle/bugfix

fix audio shape bug for audio len
4 years ago · abf7fa5518
parent 7e5fcbfb0d bb98a174cd
commit abf7fa5518
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