From bb1122188d9bd95f887e03e07d0cd602d75ca016 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Wed, 1 May 2019 11:14:21 +0200
Subject: [PATCH] Force garbage collection after every predict iteration

Signed-off-by: Jim Martens <github@2martens.de>
---
 src/twomartens/masterthesis/ssd.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index 9cadfd7..6c6fae7 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -201,6 +201,10 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
         
         counter += 1
         tf.set_random_seed(1)
+        from tensorflow.python.eager import context
+        context.context()._clear_caches()
+        import gc
+        gc.collect()
     
     epoch_end_time = time.time()
     per_epoch_time = epoch_end_time - epoch_start_time