Added parameters for batch size and ssd image size

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-04 16:30:29 +02:00
parent b6beebd0c1
commit a2fe3e860a

View File

@ -47,6 +47,10 @@ _CONFIG_PROPS = {
"Debug": { "Debug": {
"summaries": (bool, "True"), "summaries": (bool, "True"),
"train_images": (bool, "False") "train_images": (bool, "False")
},
"Parameters": {
"batch_size": (int, "32"),
"ssd_image_size": (int, "300")
} }
} }