From f14f0ae200a4d4f878d5d65a0f13ecfcfca39a9c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 19 Jun 2019 14:32:54 +0200 Subject: [PATCH] Set number of validation steps to 20 Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index f8674fc..578fe14 100644 --- 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,