From 1f85442a911934c2e9f4462f1d36aedb8170e206 Mon Sep 17 00:00:00 2001 From: Brigham Lysenko Date: Tue, 13 Aug 2019 16:45:31 -0600 Subject: [PATCH] updated log --- mainscripts/Trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 8a41c58..624d44f 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -109,7 +109,7 @@ def trainerThread (s2c, c2s, args, device_args): loss_history = model.get_loss_history() time_str = time.strftime("[%H:%M:%S]") if iter_time >= 10: - loss_string = "{0}[#{1:06d}][{2:.5s}s]".format ( time_str, iter, '{:0.4f}'.format(iter_time) ) + loss_string = "{0}[#{1:06d}][{2:.5s}s][bs: {3}]".format ( time_str, iter, '{:0.4f}'.format(iter_time), batch_size ) else: loss_string = "{0}[#{1:06d}][{2:04d}ms][bs: {3}]".format ( time_str, iter, int(iter_time*1000), batch_size)