2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added some config values

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-10 15:23:30 +02:00
parent d8d7b51655
commit b01cda512a
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/twomartens/masterthesis/config.py
View File

@ -54,7 +54,12 @@ _CONFIG_PROPS = {
"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") "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")
} }
} }