From 2b656057af5ea20bd136505882456054ef358785 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 22 Jul 2019 15:20:46 +0200 Subject: [PATCH] Enabled augmentations for training again 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 a922d51..c87a421 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -774,7 +774,7 @@ def _ssd_train_get_generators(args: argparse.Namespace, predictor_sizes=predictor_sizes, batch_size=batch_size, image_size=image_size, - training=True, evaluation=False, augment=False, + training=True, evaluation=False, augment=True, debug=args.debug, nr_trajectories=nr_trajectories)