Returned number of val batches to previous value

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-20 12:51:19 +02:00
parent 0a2857f31d
commit 0c02a94a33
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,
-        20,
+        nr_batches_val,
        ssd_model,
        weights_path,
        args.iteration,