From a2fe3e860a034d02f0021d8894638e05388dcf12 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 4 Jul 2019 16:30:29 +0200 Subject: [PATCH] Added parameters for batch size and ssd image size Signed-off-by: Jim Martens --- src/twomartens/masterthesis/config.py | 4 ++++ 1 file changed, 4 insertions(+) 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") } }