Changed mode to training for DropoutSSD to allow partitioning of observations

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-21 11:31:24 +02:00
parent 0056f28e79
commit af9cbc60d2
1 changed files with 1 additions and 1 deletions
--- 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({