From 34c754fca3bbe878f82a0450630298b145b1e697 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Wed, 8 May 2019 13:55:25 +0200
Subject: [PATCH] Added summary check to see what remains after gc

Signed-off-by: Jim Martens <github@2martens.de>
---
 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 42ad968..530e887 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -219,6 +219,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
         tf.set_random_seed(1)
         context.context()._clear_caches()
         gc.collect()
+        
+        trs.print_diff()
     
     epoch_end_time = time.time()
     per_epoch_time = epoch_end_time - epoch_start_time