From 11edf3772acf2a7314a4b538f2ed46ca9a30c0b4 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 22 May 2019 13:59:21 +0200 Subject: [PATCH] Removed extra garbage collection Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index b6b5a87..de7813a 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -184,9 +184,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset, # go through the data set counter = 0 - import gc - - from tensorflow.python.eager import context for inputs, labels in dataset: if use_dropout: @@ -248,10 +245,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset, pickle.dump(labels, label_file) counter += 1 - - tf.set_random_seed(1) - context.context()._clear_caches() - gc.collect() epoch_end_time = time.time() per_epoch_time = epoch_end_time - epoch_start_time