From f950504f20313c0ebf896a17111f22f727e85b68 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 23 Jul 2019 13:46:15 +0200 Subject: [PATCH] Added missing network selection Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 33f6ce8..406a426 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -222,7 +222,16 @@ def _build_visualise(parser: argparse.ArgumentParser) -> None: def _build_visualise_metrics(parser: argparse.ArgumentParser) -> None: - parser.add_argument("iteration", type=int, help="the validation iteration to use") + sub_parsers = parser.add_subparsers(dest="network") + sub_parsers.required = True + + ssd_bayesian_parser = sub_parsers.add_parser("bayesian_ssd", help="SSD with dropout layers") + ssd_parser = sub_parsers.add_parser("ssd", help="SSD") + auto_encoder_parser = sub_parsers.add_parser("auto_encoder", help="Auto-encoder network") + + ssd_bayesian_parser.add_argument("iteration", type=int, help="the validation iteration to use") + ssd_parser.add_argument("iteration", type=int, help="the validation iteration to use") + auto_encoder_parser.add_argument("iteration", type=int, help="the validation iteration to use") def _build_measure(parser: argparse.ArgumentParser) -> None: