Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

guided attn #8

This commit is contained in:
Eren Golge 2018-04-24 11:35:54 -07:00
parent 622987771a
commit 80c18684c6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -106,6 +106,8 @@ def train(model, criterion, data_loader, optimizer, epoch):
# create attention mask # create attention mask
# TODO: vectorize # TODO: vectorize
print(tet_input_var.shape)
print(mel_spec_var.shape)
N = text_input_var.shape[1] N = text_input_var.shape[1]
T = mel_spec_var.shape[1] T = mel_spec_var.shape[1]
M = np.zeros([N, T]) M = np.zeros([N, T])