From a2a1b9931b2e48515673bf140f36a0ef304eb670 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Wed, 25 Apr 2018 05:39:09 -0700
Subject: [PATCH] bug fix

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

diff --git a/train.py b/train.py
index 051fb8ad..a3bb8c26 100644
--- a/train.py
+++ b/train.py
@@ -124,6 +124,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
                               linear_spec_var[:, :, :n_priority_freq],
                               mel_lengths_var)
         attention_loss = criterion(alignments, M, mel_lengths_var)
+        print(mk)
         loss = mel_loss + linear_loss + mk * attention_loss
 
         # backpass and check the grad norm