From 73e4920cc4184f2f0d5ad15cfd9c708e44e7fa8c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 11 Jul 2019 15:01:49 +0200 Subject: [PATCH] Fixed mode for ssd model Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 1efb6aa..77ce8fd 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -267,7 +267,7 @@ def _ssd_test(args: argparse.Namespace) -> None: keras_ssd300.ssd_300, image_size, nr_classes, - "inference_fast", + "training", iou_threshold, dropout_rate, top_k,