diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index f8674fc..578fe14 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -95,7 +95,7 @@ def _ssd_train(args: argparse.Namespace) -> None: train_generator, nr_batches_train, val_generator, - nr_batches_val, + 20, ssd_model, weights_path, args.iteration,