From a4e6ac2853b2073d2eec0577e4cd04fd6923949b Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 11 Jun 2019 12:06:22 +0200 Subject: [PATCH] Added missing argument of num_epochs 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 2341500..8343a46 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -62,7 +62,7 @@ def _ssd_train(args: argparse.Namespace) -> None: scenenet_data, nr_digits, length_dataset = \ data.load_scenenet_data(file_names_photos, instances, args.coco_path, - batch_size=batch_size, + batch_size=batch_size, num_epochs=args.num_epochs, resized_shape=(image_size, image_size), mode="training") del file_names_photos, instances