From 446cd6fa0656c002e1f5f2a94b611ab0770ac9b5 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 14 Aug 2019 12:32:09 +0200 Subject: [PATCH] small logging bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 565713ed..233c7576 100644 --- a/train.py +++ b/train.py @@ -258,7 +258,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler, " | > EPOCH END -- GlobalStep:{} AvgTotalLoss:{:.5f} " "AvgPostnetLoss:{:.5f} AvgDecoderLoss:{:.5f} " "AvgStopLoss:{:.5f} EpochTime:{:.2f} " - "AvgStepTime:{:.2f}".format(global_step, avg_total_loss, + "AvgStepTime:{:.2f} AvgLoaderTime:{:.2f}".format(global_step, avg_total_loss, avg_postnet_loss, avg_decoder_loss, avg_stop_loss, epoch_time, avg_step_time, avg_loader_time),