From fe516f642901d0ca3fb3becf02b5e4b0079ebdf4 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 11 Jun 2019 11:34:18 +0200 Subject: [PATCH] Added additional values to print Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 00cc383..8e8f68b 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -497,7 +497,9 @@ def _train_epochs(nr_batches_per_epoch: int, print(( 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}, " + f"batch_counter: {batch_counter:d}, " + f"nr_batches_per_epoch: {nr_batches_per_epoch:d}" )) # save weights at end of epoch