Set number of validation steps to 20

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-19 14:32:54 +02:00
parent d1ee766579
commit f14f0ae200
1 changed files with 1 additions and 1 deletions
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@ -95,7 +95,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
        train_generator,
        nr_batches_train,
        val_generator,
-        nr_batches_val,
+        20,
        ssd_model,
        weights_path,
        args.iteration,