From 7932d77ee895dea39f5d174f434c856b8c575047 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 8 May 2019 13:48:59 +0200 Subject: [PATCH] Enabled summary tracker 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 77a02ab..1c17edf 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -185,8 +185,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset, from tensorflow.python.eager import context - # trs = tracker.SummaryTracker() - # trs.print_diff() + trs = tracker.SummaryTracker() + trs.print_diff() for inputs in dataset: decoded_predictions_batch = []