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

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

diff --git a/train.py b/train.py
index ff587d0e..7011b024 100644
--- a/train.py
+++ b/train.py
@@ -130,7 +130,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
                               mel_lengths_var)
         print(M.shape)
         print(alignments.shape)
-        attention_loss = criterion(M, alignments, mel_lengths_var)
+        attention_loss = criterion(alignments, M, mel_lengths_var)
         loss = mel_loss + linear_loss + 0.2 * attention_loss
 
         # backpass and check the grad norm