guided attn #8

pull/10/head
Eren Golge 2018-04-24 11:35:54 -07:00
parent 622987771a
commit 80c18684c6
1 changed files with 2 additions and 0 deletions

View File

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