bug fix

2018-04-25 05:37:21 -07:00 · 2018-04-25 05:37:21 -07:00 · 3c4dcedbd7
parent d61a79db89
commit 3c4dcedbd7
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -111,7 +111,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
        # create attention mask
        N = text_input_var.shape[1]
        T = mel_spec_var.shape[1] // c.r
-        M = create_attn_mask(N, T, g)
+        M = create_attn_mask(N, T, 0.03)
        
        # forward pass
        mel_output, linear_output, alignments =\