From af9cbc60d26686f93f656941aafc4924a4cdd6d4 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Tue, 21 May 2019 11:31:24 +0200
Subject: [PATCH] Changed mode to training for DropoutSSD to allow partitioning
 of observations

Signed-off-by: Jim Martens <github@2martens.de>
---
 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 80d9e08..f977789 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -141,7 +141,7 @@ def predict(dataset: tf.data.Dataset,
     checkpointables = {}
     if use_dropout:
         checkpointables.update({
-            'ssd': DropoutSSD(mode='inference_fast', weights_path=weights_path)
+            'ssd': DropoutSSD(mode='training', weights_path=weights_path)
         })
     else:
         checkpointables.update({