Added option "ssd_forward_passes_per_image"

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-07 13:50:59 +02:00
parent 4886fda32e
commit 3e9c90f201
1 changed files with 2 additions and 1 deletions
--- a/src/twomartens/masterthesis/config.py
+++ b/src/twomartens/masterthesis/config.py
@ -52,7 +52,8 @@ _CONFIG_PROPS = {
        "batch_size": (int, "32"),
        "ssd_image_size": (int, "300"),
        "learning_rate": (float, "0.001"),
-        "steps_per_val_epoch": (int, "20")
+        "steps_per_val_epoch": (int, "20"),
+        "ssd_forward_passes_per_image": (int, "10")
    }
 }