Skip to content

Commit be9162e

Browse files
MILK-BIOSPillar1989
authored andcommitted
Fix: Fix wrong import of inference and export
1 parent 35a3862 commit be9162e

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tools/export.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -483,11 +483,11 @@ def main():
483483
args, cfg = build_config(args)
484484

485485
if 'runner_type' not in cfg:
486-
from mmengine.runner import Runner
486+
from sscma.engine import Runner
487487

488488
runner = Runner.from_cfg(cfg)
489489
else:
490-
from mmengine.registry import RUNNERS
490+
from sscma.registry import RUNNERS
491491

492492
runner = RUNNERS.build(cfg)
493493

tools/inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -280,11 +280,11 @@ def main():
280280
args, cfg = build_config(args)
281281

282282
if 'runner_type' not in cfg:
283-
from mmengine.runner import Runner
283+
from sscma.engine import Runner
284284

285285
runner = Runner.from_cfg(cfg)
286286
else:
287-
from mmengine.registry import RUNNERS
287+
from sscma.registry import RUNNERS
288288

289289
runner = RUNNERS.build(cfg)
290290

0 commit comments

Comments
 (0)