From 5c8e66600e3cad7e1225e3fa5cabb4a448b88392 Mon Sep 17 00:00:00 2001 From: gongel Date: Wed, 20 Oct 2021 11:04:32 +0000 Subject: [PATCH] fix: bugfix --- deepspeech/io/collator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepspeech/io/collator.py b/deepspeech/io/collator.py index ae1feb78..5f0bc462 100644 --- a/deepspeech/io/collator.py +++ b/deepspeech/io/collator.py @@ -194,7 +194,7 @@ class SpeechCollatorBase(): texts = [] text_lens = [] utts = [] - tids = [] # tokenids + tids = [] # tokenids for idx, item in enumerate(batch): utts.append(item['utt']) @@ -354,7 +354,7 @@ class TripletSpeechCollator(SpeechCollator): tokens = [[], []] for idx, text in enumerate([translation, transcription]): - tokens[idx] = tokenids(text, self.keep_transcription_text) + tokens[idx] = _tokenids(text, self.keep_transcription_text) translation_text.append(tokens[0]) translation_text_lens.append(tokens[0].shape[0])