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