mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add argument to override configs (#91)

pull/96/head
LXXXXR 2020-11-20 13:50:28 +08:00 committed by GitHub
parent 149ee3a30d
commit e75f2b7c35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/test.py
View File

@ -5,6 +5,7 @@ import warnings
import mmcv import mmcv
import numpy as np import numpy as np
import torch import torch
from mmcv import DictAction
from mmcv.parallel import MMDataParallel, MMDistributedDataParallel from mmcv.parallel import MMDataParallel, MMDistributedDataParallel
from mmcv.runner import get_dist_info, init_dist, load_checkpoint from mmcv.runner import get_dist_info, init_dist, load_checkpoint
@ -26,6 +27,12 @@ def parse_args():
action='store_true', action='store_true',
help='whether to use gpu to collect results') help='whether to use gpu to collect results')
parser.add_argument('--tmpdir', help='tmp dir for writing some results') parser.add_argument('--tmpdir', help='tmp dir for writing some results')
parser.add_argument(
'--options',
nargs='+',
action=DictAction,
help='override some settings in the used config, the key-value pair '
'in xxx=yyy format will be merged into config file.')
parser.add_argument( parser.add_argument(
'--launcher', '--launcher',
choices=['none', 'pytorch', 'slurm', 'mpi'], choices=['none', 'pytorch', 'slurm', 'mpi'],
@ -42,6 +49,8 @@ def main():
args = parse_args() args = parse_args()
cfg = mmcv.Config.fromfile(args.config) cfg = mmcv.Config.fromfile(args.config)
if args.options is not None:
cfg.merge_from_dict(args.options)
# set cudnn_benchmark # set cudnn_benchmark
if cfg.get('cudnn_benchmark', False): if cfg.get('cudnn_benchmark', False):
torch.backends.cudnn.benchmark = True torch.backends.cudnn.benchmark = True