From eef60785339781f16535aa2a5753e534a9c820bd Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 11 Jun 2019 12:00:08 +0200 Subject: [PATCH] Added print to show epoch number This can be used to identify if the data set is in fact repeated. In that case this print line has to be printed at least twice. Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 8e8f68b..84e2429 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -465,6 +465,10 @@ def _train_epochs(nr_batches_per_epoch: int, # epoch starts epoch_start_time = time.time() ssd_loss_avg = tfe.metrics.Mean(name='ssd_loss', dtype=tf.float32) + if verbose: + print(( + f"epoch: {epoch + 1:d}" + )) labels = [] for i in range(y.shape[0]):