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

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

diff --git a/train.py b/train.py
index 6b56ed59..ff587d0e 100644
--- a/train.py
+++ b/train.py
@@ -107,7 +107,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
         # create attention mask
         # TODO: vectorize
         N = text_input_var.shape[1]
-        T = mel_spec_var.shape[1] / c.r
+        T = mel_spec_var.shape[1] // c.r
         M = np.zeros([N, T])
         for t in range(T):
             for n in range(N):