Fixed mode for ssd model

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:01:49 +02:00 · 2019-07-11 15:01:49 +02:00 · 73e4920cc4
parent fef7a1eac1
commit 73e4920cc4
1 changed files with 1 additions and 1 deletions
--- 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,