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

Changed train function to be an integration 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-10 12:15:17 +02:00
parent a90c6c7c92
commit 91b7febabb
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/twomartens/masterthesis/cli.py
View File

@ -58,10 +58,7 @@ def prepare(args: argparse.Namespace) -> None:
def train(args: argparse.Namespace) -> None:
if args.network == "ssd" or args.network == "bayesian_ssd":
_ssd_train(args)
elif args.network == "auto_encoder":
_auto_encoder_train(args)
_train_execute_action(args, _ssd_train, _auto_encoder_train)
def test(args: argparse.Namespace) -> None:
@ -164,6 +161,13 @@ def _config_execute_action(args: argparse.Namespace, on_get: callable,
on_list()
def _train_execute_action(args: argparse.Namespace, on_ssd: callable, on_auto_encoder: callable) -> None:
if args.network == "ssd" or args.network == "bayesian_ssd":
on_ssd(args)
elif args.network == "auto_encoder":
on_auto_encoder(args)
def _ssd_train(args: argparse.Namespace) -> None:
import os
import pickle