From a42669327e9ff433bbe6c1e16027b5e2985ed785 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 24 Apr 2018 11:43:24 -0700 Subject: [PATCH] guided attn #11 --- train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/train.py b/train.py index 7011b024..7b5a02fe 100644 --- a/train.py +++ b/train.py @@ -128,8 +128,6 @@ 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) - print(M.shape) - print(alignments.shape) attention_loss = criterion(alignments, M, mel_lengths_var) loss = mel_loss + linear_loss + 0.2 * attention_loss