From 19e6f4a8b76a25328337f8f31ed73bdb764f8d84 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 26 Apr 2018 05:47:15 -0700 Subject: [PATCH] bug fix --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 668c9162..b58b513a 100644 --- a/train.py +++ b/train.py @@ -121,7 +121,7 @@ def train(model, criterion, data_loader, optimizer, epoch): mel_loss = criterion(mel_output, mel_spec, mel_lengths) linear_loss = criterion(linear_output, linear_spec, mel_lengths) if c.priority_freq: - linear_loss = 0.5 * linear_loss + linear_loss = 0.5 * linear_loss\ + 0.5 * criterion(linear_output[:, :, :n_priority_freq], linear_spec[:, :, :n_priority_freq], mel_lengths) @@ -248,7 +248,7 @@ def evaluate(model, criterion, data_loader, current_step): mel_loss = criterion(mel_output, mel_spec, mel_lengths) linear_loss = criterion(linear_output, linear_spec, mel_lengths) if c.priority_freq: - linear_loss = 0.5 * linear_loss + linear_loss = 0.5 * linear_loss\ + 0.5 * criterion(linear_output[:, :, :n_priority_freq], linear_spec[:, :, :n_priority_freq], mel_lengths)