From b01cda512ac2418d06f80ed7eb12dec0843bc9a7 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Wed, 10 Jul 2019 15:23:30 +0200
Subject: [PATCH] Added some config values

Signed-off-by: Jim Martens <github@2martens.de>
---
 src/twomartens/masterthesis/config.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/twomartens/masterthesis/config.py b/src/twomartens/masterthesis/config.py
index 2e082c4..c29ff61 100644
--- 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")
     }
 }