From 62ee6094f1e78c2ed29c8abfa1f49361aeb79a8a Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 8 May 2019 14:27:07 +0200 Subject: [PATCH] Removed memory tracking code Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 1ad15a9..b464bdf 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -181,13 +181,9 @@ 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 - trs = tracker.SummaryTracker() - orig_summary = trs.create_summary() - for inputs in dataset: decoded_predictions_batch = [] if use_dropout: @@ -197,8 +193,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset, del result else: result = np.array(ssd(inputs)) - print("After network call") - trs.print_diff(orig_summary) decoded_predictions_batch.append(result) del result