bug fix

2019-04-30 10:59:29 +02:00 · 2019-04-30 10:59:29 +02:00 · afb5a17221
parent 8f6721a1ff
commit afb5a17221
1 changed files with 1 additions and 0 deletions
--- 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