From 530ec99feb347a5b3a2b75b029b4df1af3618d0c Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 24 Apr 2018 11:31:25 -0700 Subject: [PATCH] guided attn #5 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 7c47c394..7756ab37 100644 --- a/train.py +++ b/train.py @@ -128,7 +128,7 @@ def train(model, criterion, data_loader, optimizer, epoch): + 0.5 * criterion(linear_output[:, :, :n_priority_freq], linear_spec_var[:, :, :n_priority_freq], mel_lengths_var) - attention_loss = criterion(Mg, alignments, mel_lengths_var) + attention_loss = criterion(M, alignments, mel_lengths_var) loss = mel_loss + linear_loss + 0.2 * attention_loss # backpass and check the grad norm