Added some config values

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:23:30 +02:00
parent d8d7b51655
commit b01cda512a
1 changed files with 6 additions and 1 deletions
--- a/src/twomartens/masterthesis/config.py
+++ b/src/twomartens/masterthesis/config.py
@ -54,7 +54,12 @@ _CONFIG_PROPS = {
        "ssd_image_size": (int, "300"),
        "learning_rate": (float, "0.001"),
        "steps_per_val_epoch": (int, "20"),
-        "ssd_forward_passes_per_image": (int, "10")
+        "ssd_forward_passes_per_image": (int, "10"),
+        "nr_classes": (int, "80"),
+        "ssd_iou_threshold": (float, "0.45"),
+        "ssd_top_k": (int, "200"),
+        "ssd_dropout_rate": (float, "0.5"),
+        "nr_trajectories": (int, "-1")
    }
 }