mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[Enhancement] Support custom runner (#1923)

Browse Source 
* support custom runner

* fix
This commit is contained in:
谢昕辰 2022-08-17 13:46:57 +08:00 committed by MeowZheng
parent 45b63c584f
commit 13b108dbc8
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/train.py
View File

@ -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():
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()