From 1bcaf44e48069b2b9a40c33be07b5813243fb21c Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 24 Apr 2018 11:28:40 -0700 Subject: [PATCH] guided attn #2 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 8dd6a192..a53015c0 100644 --- a/train.py +++ b/train.py @@ -126,7 +126,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(, 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