From 97ffa2b44ed81a7f892b45e263e2029f4e401169 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 22 Aug 2019 15:28:46 +0200 Subject: [PATCH] indent --- train.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/train.py b/train.py index 71cbb398..1f808e49 100644 --- a/train.py +++ b/train.py @@ -215,11 +215,11 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler, # reduce TB load if global_step % 10 == 0: iter_stats = {"loss_posnet": postnet_loss.item(), - "loss_decoder": decoder_loss.item(), - "lr": current_lr, - "grad_norm": grad_norm, - "grad_norm_st": grad_norm_st, - "step_time": step_time} + "loss_decoder": decoder_loss.item(), + "lr": current_lr, + "grad_norm": grad_norm, + "grad_norm_st": grad_norm_st, + "step_time": step_time} tb_logger.tb_train_iter_stats(global_step, iter_stats) if global_step % c.save_step == 0: