From 80c18684c6df3c09a92bfd2c8a1a6e87c7453dea Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Tue, 24 Apr 2018 11:35:54 -0700
Subject: [PATCH] guided attn #8

---
 train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/train.py b/train.py
index a12e1f46..08a1f5a2 100644
--- 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])