Skip to content
Snippets Groups Projects
Commit 13b108db authored by 谢昕辰's avatar 谢昕辰 Committed by MeowZheng
Browse files

[Enhancement] Support custom runner (#1923)

* support custom runner

* fix
parent 45b63c58
No related branches found
No related tags found
No related merge requests found
...@@ -6,6 +6,7 @@ import os.path as osp ...@@ -6,6 +6,7 @@ import os.path as osp
from mmengine.config import Config, DictAction from mmengine.config import Config, DictAction
from mmengine.logging import print_log from mmengine.logging import print_log
from mmengine.registry import RUNNERS
from mmengine.runner import Runner from mmengine.runner import Runner
from mmseg.utils import register_all_modules from mmseg.utils import register_all_modules
...@@ -97,7 +98,13 @@ def main(): ...@@ -97,7 +98,13 @@ def main():
cfg.load_from = args.resume cfg.load_from = args.resume
# build the runner from config # 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 training # start training
runner.train() runner.train()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment