Merge pull request #104 from iclementine/release/v0.2

update code for tacotron2
This commit is contained in:
Feiyu Chan 2021-04-13 16:19:09 +08:00 committed by GitHub
commit 66620977ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -86,6 +86,7 @@ class LJSpeechCollector(object):
for i, _ in sorted(
zip(mel_lens, text_lens), key=lambda x: x[1], reverse=True)
]
mel_lens = np.array(mel_lens, dtype=np.int64)
stop_tokens = [
i
@ -93,7 +94,7 @@ class LJSpeechCollector(object):
zip(stop_tokens, text_lens), key=lambda x: x[1], reverse=True)
]
text_lens = sorted(text_lens, reverse=True)
text_lens = np.array(sorted(text_lens, reverse=True), dtype=np.int64)
# Pad sequence with largest len of the batch
texts = batch_text_id(texts, pad_id=self.padding_idx)