Added missing argument of num_epochs

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-11 12:06:22 +02:00
parent 057bffcae3
commit a4e6ac2853
1 changed files with 1 additions and 1 deletions
--- 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