bug fix

2026-09-08 16:50:36 +08:00 · 2018-04-25 05:37:21 -07:00
parent 8d85d01c8a
commit 03033a99ef
1 changed files with 1 additions and 1 deletions
@@ -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 =\