plot attention alignments

This commit is contained in:
Eren Golge
2018-02-02 05:37:09 -08:00
parent 235ce071c6
commit 4e0ab65bbf
5 changed files with 66 additions and 55 deletions
Binary file not shown.
+1 -1
View File
@@ -53,7 +53,7 @@ class LJSpeechDataset(Dataset):
text = [d['text'] for d in batch]
text_lenghts = [len(x) for x in text]
max_text_len = np.max(text_lengths)
max_text_len = np.max(text_lenghts)
wav = [d['wav'] for d in batch]
# PAD sequences with largest length of the batch