From 2b64209bda28eb5cbec09a94be6fd34d3b414857 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 17 Jun 2019 18:20:02 +0200 Subject: [PATCH] Added missing function argument Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 50dbd4f..9db87f4 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -75,13 +75,13 @@ def _ssd_train(args: argparse.Namespace) -> None: predictor_sizes=ssd_model.predictor_sizes, batch_size=batch_size, resized_shape=(image_size, image_size), - training=True) + training=True, evaluation=False) val_generator, val_length = \ data.load_scenenet_data(file_names_val, instances_val, args.coco_path, predictor_sizes=ssd_model.predictor_sizes, batch_size=batch_size, resized_shape=(image_size, image_size), - training=False) + training=False, evaluation=False) del file_names_train, instances_train, file_names_val, instances_val nr_batches_train = int(math.ceil(train_length / float(batch_size)))