From 8770d5d7310d9c15c31db2a3a805e3f5e59988a7 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 16:07:31 +0200 Subject: [PATCH] Improved location of tracker initialization Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index c47ee13..55a74ab 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -181,9 +181,12 @@ def _predict_one_epoch(dataset: tf.data.Dataset, # go through the data set counter = 0 import gc + from pympler import tracker + from tensorflow.python.eager import context - tr = None - trs = None + + tr = tracker.ObjectTracker() + trs = tracker.SummaryTracker() for inputs in dataset: decoded_predictions_batch = [] @@ -215,16 +218,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset, context.context()._clear_caches() gc.collect() - from pympler import tracker - if tr is None: - tr = tracker.ObjectTracker() - else: - tr.print_diff() - - if trs is None: - trs = tracker.SummaryTracker() - else: - trs.print_diff() + tr.print_diff() + trs.print_diff() epoch_end_time = time.time() per_epoch_time = epoch_end_time - epoch_start_time