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