guided attn #8

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