Merge pull request #915 from gongel/dp_bugfix

[BUGFIX]  tokenids ---> _tokenids
pull/1091/head
Hui Zhang 4 years ago committed by GitHub
commit dccae6f72b

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

Loading…
Cancel
Save