diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index b2291f6..15e83c9 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='training', weights_path=weights_path)
+            'ssd': SSD(mode='inference_fast', weights_path=weights_path)
         })
 
     if checkpoint_path is not None:
diff --git a/src/twomartens/masterthesis/ssd_keras b/src/twomartens/masterthesis/ssd_keras
index ccf74fe..0cb1f77 160000
--- a/src/twomartens/masterthesis/ssd_keras
+++ b/src/twomartens/masterthesis/ssd_keras
@@ -1 +1 @@
-Subproject commit ccf74fe224cb28c62f9d6d6e5e8c919ff7ed81c9
+Subproject commit 0cb1f7750e76fa6b21dad9ec38eb8fb1cdca0450