From afb5a17221a2cfc48a808f1dee8c5612320cdf93 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Tue, 30 Apr 2019 10:59:29 +0200
Subject: [PATCH] bug fix

---
 layers/tacotron2.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/layers/tacotron2.py b/layers/tacotron2.py
index 2fa6d06f..0826ccc6 100644
--- a/layers/tacotron2.py
+++ b/layers/tacotron2.py
@@ -251,6 +251,7 @@ class Attention(nn.Module):
         else:
             context = torch.bmm(alignment.unsqueeze(1), inputs)
             context = context.squeeze(1)
+            self.attention_weights = alignment
         return context