Added parameters for batch size and ssd image size
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -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")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user