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])