From 40b479b3b95eed469c4a1f5c6ea96609707f4542 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 30 Apr 2018 06:11:22 -0700 Subject: [PATCH] bug fix --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 5805ea5c..de3bfe9c 100644 --- a/train.py +++ b/train.py @@ -115,7 +115,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, epoch): # loss computation mel_loss = criterion(mel_output, mel_spec, mel_lengths) linear_loss = criterion(linear_output, linear_spec, mel_lengths) - stop_loss = criterion_st(stop_tokens, stop_targets) + stop_loss = criterion_st(stop_tokens, stop_target) if c.priority_freq: linear_loss = 0.5 * linear_loss\ + 0.5 * criterion(linear_output[:, :, :n_priority_freq], @@ -249,7 +249,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step): # loss computation mel_loss = criterion(mel_output, mel_spec, mel_lengths) linear_loss = criterion(linear_output, linear_spec, mel_lengths) - stop_loss = criterion_st(stop_tokens, stop_targets) + stop_loss = criterion_st(stop_tokens, stop_target) if c.priority_freq: linear_loss = 0.5 * linear_loss\ + 0.5 * criterion(linear_output[:, :, :n_priority_freq],