guided attn #9

2018-04-24 11:36:56 -07:00 · 2018-04-24 11:36:56 -07:00 · ee00b3db5c
parent 9bfcdd1cf5
commit ee00b3db5c
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -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]