From 616188b19f6cd1467e958519e42c1bd1267709d7 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 16:12:57 +0200 Subject: [PATCH] Figure out impact of garbage collection Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 3f9775b..3b09d3b 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -212,6 +212,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset, np.save(file, decoded_predictions_batch_np, allow_pickle=False, fix_imports=False) counter += 1 + + trs.print_diff() tf.set_random_seed(1) context.context()._clear_caches()