File size: 871 Bytes
491eded |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import argparse
import os.path as osp
from datetime import datetime
import pytz
def default_argument_parser(add_help=True, default_config_file=""):
parser = argparse.ArgumentParser(add_help=add_help)
parser.add_argument("--config-file", '-c', default=default_config_file, metavar="FILE", help="path to config file")
parser.add_argument(
"--opts",
help="Modify config options using the command-line",
default=None,
nargs=argparse.REMAINDER,
)
return parser
def setup(args, freeze=True):
from .defaults import _C as cfg
cfg = cfg.clone()
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
dt = datetime.now(pytz.timezone('America/Los_Angeles')).strftime('%y%m%d-%H%M%S')
cfg.output_dir = osp.join(cfg.output_dir, cfg.name, dt)
if freeze:
cfg.freeze()
return cfg |