guided attn #11

2018-04-24 11:39:58 -07:00 · 2018-04-24 11:39:58 -07:00 · e87d7d8d26
parent 830a051a78
commit e87d7d8d26
1 changed files with 1 additions and 1 deletions
--- 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):