fix: bugfix

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

@ -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