diff --git a/src/twomartens/masterthesis/config.py b/src/twomartens/masterthesis/config.py index c793ee3..994a56c 100644 --- a/src/twomartens/masterthesis/config.py +++ b/src/twomartens/masterthesis/config.py @@ -47,6 +47,10 @@ _CONFIG_PROPS = { "Debug": { "summaries": (bool, "True"), "train_images": (bool, "False") + }, + "Parameters": { + "batch_size": (int, "32"), + "ssd_image_size": (int, "300") } }