From 8ae1ef72fc21b5ad0f1312413768bdcc82af0547 Mon Sep 17 00:00:00 2001 From: Eren G Date: Wed, 18 Jul 2018 14:31:09 +0200 Subject: [PATCH] logging change --- utils/generic_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/generic_utils.py b/utils/generic_utils.py index a5393617..b16c6944 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -82,7 +82,7 @@ def save_checkpoint(model, optimizer, model_loss, out_path, current_step, epoch): checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step) checkpoint_path = os.path.join(out_path, checkpoint_path) - print("\n | > Checkpoint saving : {}".format(checkpoint_path)) + print(" | | > Checkpoint saving : {}".format(checkpoint_path)) new_state_dict = _trim_model_state_dict(model.state_dict()) state = {'model': new_state_dict,