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

Added additional hyper parameters to config

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 17:11:13 +02:00
parent da2b788447
commit 77e86f85e4
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/cli.py
View File

@ -239,13 +239,13 @@ def _ssd_train(args: argparse.Namespace) -> None:
train_generator, train_generator,
nr_batches_train, nr_batches_train,
val_generator, val_generator,
20, conf.get_property("Parameters.steps_per_val_epoch"),
ssd_model, ssd_model,
weights_path, weights_path,
args.iteration, args.iteration,
initial_epoch=0, initial_epoch=0,
nr_epochs=args.num_epochs, nr_epochs=args.num_epochs,
lr=0.001, lr=conf.get_property("Parameters.learning_rate"),
tensorboard_callback=tensorboard_callback tensorboard_callback=tensorboard_callback
) )

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

@ -50,7 +50,9 @@ _CONFIG_PROPS = {
}, },
"Parameters": { "Parameters": {
"batch_size": (int, "32"), "batch_size": (int, "32"),
"ssd_image_size": (int, "300") "ssd_image_size": (int, "300"),
"learning_rate": (float, "0.001"),
"steps_per_val_epoch": (int, "20")
} }
} }