From f6f6a88c50a5aaa97ed398357a6d7ffc28cd46ca Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 11 Jun 2019 11:31:04 +0200 Subject: [PATCH] Improve verbose printing for training Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index f74d7fa..4cd3432 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -498,9 +498,9 @@ def _train_epochs(nr_batches_per_epoch: int, if verbose: print(( - f"[{epoch + 1:d}/{nr_epochs:d} - " + f"[{epoch + 1:d}/{nr_epochs:d}] - " f"train time: {outputs['per_epoch_time']:.2f}, " - f"SSD loss: {outputs['ssd_loss']:.3f}, " + f"SSD loss: {outputs['ssd_loss']:.3f}" )) # save weights at end of epoch