Fixed mode of native SSD for val

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-10 14:17:50 +02:00
parent 9dbfa5c640
commit 15498549b7
1 changed files with 1 additions and 1 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -175,7 +175,7 @@ def predict(dataset: tf.data.Dataset,
    if use_dropout:
        ssd = DropoutSSD(mode='training', weights_path=weights_path)
    else:
-        ssd = SSD(mode='training', weights_path=weights_path)
+        ssd = SSD(mode='inference_fast', weights_path=weights_path)
    
    checkpointables = {
        'ssd': ssd.model,