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") } }