wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-27 12:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

guided attn #9

Browse Source
This commit is contained in:
Eren Golge committed 2018-04-24 11:36:56 -07:00
1 parent 80c18684c6
commit 8c4e03cebf
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
View File
@@ -106,7 +106,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
# create attention mask
# TODO: vectorize
print(tet_input_var.shape)
print(text_input_var.shape)
print(mel_spec_var.shape)
N = text_input_var.shape[1]
T = mel_spec_var.shape[1]