From 22dcc4f7d0a5f6f4eaa09b457d100e2084ea106e Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 22 Nov 2018 17:03:53 +0100 Subject: [PATCH] small print formatting --- 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 c56c3edf..86fe318c 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -123,7 +123,7 @@ def save_best_model(model, optimizer, model_loss, best_loss, out_path, best_loss = model_loss bestmodel_path = 'best_model.pth.tar' bestmodel_path = os.path.join(out_path, bestmodel_path) - print(" | > Best model saving with loss {0:.2f} : {1:}".format( + print(" | > Best model saving with loss {0:.5f} : {1:}".format( model_loss, bestmodel_path)) torch.save(state, bestmodel_path) return best_loss