2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added missing network selection

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-23 13:46:15 +02:00
parent 53472a8342
commit f950504f20
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/twomartens/masterthesis/main.py
View File

@ -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: