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

Added train iteration argument to CLI

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-10 11:27:10 +02:00
parent e602e7339d
commit 9dbfa5c640
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -242,7 +242,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
use_dropout = False if args.network == "ssd" else True
weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5"
checkpoint_path = f"{args.weights_path}/train/{args.network}/"
checkpoint_path = f"{args.weights_path}/train/{args.network}/{args.train_iteration}"
output_path = f"{args.output_path}/val/{args.network}/{args.iteration}/"
os.makedirs(output_path, exist_ok=True)

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

@ -121,6 +121,7 @@ def _build_ssd_val(parser: argparse.ArgumentParser) -> None:
parser.add_argument("--summary_path", type=str, help="path to the summaries directory")
parser.add_argument("--output_path", type=str, help="path to the output directory")
parser.add_argument("iteration", type=int, help="the validation iteration")
parser.add_argument("train_iteration", type=int, help="the train iteration")
def _build_auto_encoder_val(parser: argparse.ArgumentParser) -> None: