From 03033a99efc67c07e95de006f6b483be040ab72d Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 25 Apr 2018 05:37:21 -0700 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 2927dd88..051fb8ad 100644 --- 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 =\