Added some config values

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-10 15:23:30 +02:00
parent d8d7b51655
commit b01cda512a

View File

@ -54,7 +54,12 @@ _CONFIG_PROPS = {
"ssd_image_size": (int, "300"),
"learning_rate": (float, "0.001"),
"steps_per_val_epoch": (int, "20"),
"ssd_forward_passes_per_image": (int, "10")
"ssd_forward_passes_per_image": (int, "10"),
"nr_classes": (int, "80"),
"ssd_iou_threshold": (float, "0.45"),
"ssd_top_k": (int, "200"),
"ssd_dropout_rate": (float, "0.5"),
"nr_trajectories": (int, "-1")
}
}