Added option "ssd_forward_passes_per_image"
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -52,7 +52,8 @@ _CONFIG_PROPS = {
|
|||||||
"batch_size": (int, "32"),
|
"batch_size": (int, "32"),
|
||||||
"ssd_image_size": (int, "300"),
|
"ssd_image_size": (int, "300"),
|
||||||
"learning_rate": (float, "0.001"),
|
"learning_rate": (float, "0.001"),
|
||||||
"steps_per_val_epoch": (int, "20")
|
"steps_per_val_epoch": (int, "20"),
|
||||||
|
"ssd_forward_passes_per_image": (int, "10")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user