From 280dbbc878c0fa122a4dbc6aae018c513883783c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 15:07:43 +0200 Subject: [PATCH] Changed SSD to training mode to test origin of memory leak Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 15e83c9..b2291f6 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -144,7 +144,7 @@ def predict(dataset: tf.data.Dataset, }) else: checkpointables.update({ - 'ssd': SSD(mode='inference_fast', weights_path=weights_path) + 'ssd': SSD(mode='training', weights_path=weights_path) }) if checkpoint_path is not None: