diff --git a/tools/test.py b/tools/test.py index 0d7f39b3a8..5a6c7be603 100644 --- a/tools/test.py +++ b/tools/test.py @@ -5,6 +5,7 @@ from mmengine.config import Config, DictAction from mmengine.runner import Runner +from mmseg.registry import RUNNERS # TODO: support fuse_conv_bn, visualization, and format_only @@ -113,7 +114,13 @@ def main(): cfg.test_evaluator['keep_results'] = True # build the runner from config - runner = Runner.from_cfg(cfg) + if 'runner_type' not in cfg: + # build the default runner + runner = Runner.from_cfg(cfg) + else: + # build customized runner from the registry + # if 'runner_type' is set in the cfg + runner = RUNNERS.build(cfg) # start testing runner.test()