From 15498549b713b04f47df833ba85bd4272dd83d79 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 10 Jun 2019 14:17:50 +0200 Subject: [PATCH] Fixed mode of native SSD for val 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 729a9c0..658b142 100644 --- 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,