Merge pull request #691 from PaddlePaddle/bugfix

fix audio shape bug for audio len
3 years ago · d1c280cce8
parent dd433982e0 9c0b6c5bb0
commit d1c280cce8
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