fix: bugfix

pull/915/head
gongel 4 years ago
parent 0674d9ec24
commit 5c8e66600e

@ -194,7 +194,7 @@ class SpeechCollatorBase():
texts = [] texts = []
text_lens = [] text_lens = []
utts = [] utts = []
tids = [] # tokenids tids = [] # tokenids
for idx, item in enumerate(batch): for idx, item in enumerate(batch):
utts.append(item['utt']) utts.append(item['utt'])
@ -354,7 +354,7 @@ class TripletSpeechCollator(SpeechCollator):
tokens = [[], []] tokens = [[], []]
for idx, text in enumerate([translation, transcription]): 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.append(tokens[0])
translation_text_lens.append(tokens[0].shape[0]) translation_text_lens.append(tokens[0].shape[0])

Loading…
Cancel
Save