-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.py
21 lines (16 loc) · 795 Bytes
/
run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import argparse
import importlib
from pathlib import Path
from utils import load_args
parser = argparse.ArgumentParser(description="Semantic Segmentation Training")
parser.add_argument('--script-dir', type=str, required=True, help='Indicate directory for main file')
parser.add_argument('--config-file', type=str, required=True, help='Path of JSON config file')
parser.add_argument('--eval', type=str, help='Run evaluation with model at given path')
args = parser.parse_args()
config_file = Path(__file__).parent / 'scripts' / args.script_dir / args.config_file
train_args = load_args(config_file)
if args.eval is not None:
train_args.resume = args.eval
train_args.eval_only = True
module = importlib.import_module("scripts.{}.main".format(args.script_dir))
module.main(train_args)