From bd499425edf0f84e3cadc32264ff8ddbf3d90f2a Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 4 Jul 2019 13:15:51 +0200 Subject: [PATCH] Only use one trajectory for training and validation data each Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 74ac680..cb74b71 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -75,13 +75,15 @@ 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, evaluation=False, augment=False) + training=True, evaluation=False, augment=False, + nr_trajectories=1) 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, evaluation=False, augment=False) + training=False, evaluation=False, augment=False, + nr_trajectories=1) del file_names_train, instances_train, file_names_val, instances_val if args.debug: