diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 65c4759..250aa94 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -31,7 +31,7 @@ def main() -> None: description="Train, test, and use SSD with novelty detection.", ) - parser.add_argument("--verbose", action="store_true", help="provide to get extra output") + parser.add_argument("--verbose", default=False, action="store_true", help="provide to get extra output") parser.add_argument('--version', action='version', version='2martens Masterthesis 0.1.0') sub_parsers = parser.add_subparsers(dest="action") sub_parsers.required = True @@ -111,7 +111,7 @@ def _auto_encoder_train(args: argparse.Namespace) -> None: with train_summary_writer.as_default(): train.train_simple(coco_data, iteration=args.iteration, weights_prefix=f"{args.weights_path}/category-{category}", - zsize=64, lr=0.0001, + zsize=64, lr=0.0001, verbose=args.verbose, channels=3, train_epoch=args.num_epochs, batch_size=batch_size)