diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index de59937..c131fc5 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -239,13 +239,13 @@ def _ssd_train(args: argparse.Namespace) -> None: train_generator, nr_batches_train, val_generator, - 20, + conf.get_property("Parameters.steps_per_val_epoch"), ssd_model, weights_path, args.iteration, initial_epoch=0, nr_epochs=args.num_epochs, - lr=0.001, + lr=conf.get_property("Parameters.learning_rate"), tensorboard_callback=tensorboard_callback ) diff --git a/src/twomartens/masterthesis/config.py b/src/twomartens/masterthesis/config.py index 5a50b69..9df16ab 100644 --- a/src/twomartens/masterthesis/config.py +++ b/src/twomartens/masterthesis/config.py @@ -50,7 +50,9 @@ _CONFIG_PROPS = { }, "Parameters": { "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") } }