From cc26cd6c7b6426a2e8a82a59238f57dde24793a1 Mon Sep 17 00:00:00 2001 From: Eren G Date: Mon, 9 Jul 2018 16:11:14 +0200 Subject: [PATCH] Logging bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 01ed1bd4..040fc93e 100644 --- a/train.py +++ b/train.py @@ -320,7 +320,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step): avg_linear_loss /= (num_iter + 1) avg_mel_loss /= (num_iter + 1) avg_stop_loss /= (num_iter + 1) - avg_total_loss = avg_mel_loss + avg_linear_loss + stop_loss + avg_total_loss = avg_mel_loss + avg_linear_loss + avg_stop_loss # Plot Learning Stats tb.add_scalar('ValEpochLoss/TotalLoss', avg_total_loss, current_step)