From c4c1d94a60629462b3fd5968152a4ab39f793e0f Mon Sep 17 00:00:00 2001
From: CHENG XIN <52120610+1921134176@users.noreply.github.com>
Date: Thu, 21 Nov 2024 09:37:21 +0800
Subject: [PATCH] Update test.py

Fix fsdp weight reading error
---
 tools/test.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

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