diff --git a/tools/export.py b/tools/export.py index bf22effb..53b5ab48 100644 --- a/tools/export.py +++ b/tools/export.py @@ -483,11 +483,11 @@ def main(): args, cfg = build_config(args) if 'runner_type' not in cfg: - from mmengine.runner import Runner + from sscma.engine import Runner runner = Runner.from_cfg(cfg) else: - from mmengine.registry import RUNNERS + from sscma.registry import RUNNERS runner = RUNNERS.build(cfg) diff --git a/tools/inference.py b/tools/inference.py index 00430aba..a9ec5d1a 100644 --- a/tools/inference.py +++ b/tools/inference.py @@ -280,11 +280,11 @@ def main(): args, cfg = build_config(args) if 'runner_type' not in cfg: - from mmengine.runner import Runner + from sscma.engine import Runner runner = Runner.from_cfg(cfg) else: - from mmengine.registry import RUNNERS + from sscma.registry import RUNNERS runner = RUNNERS.build(cfg)