bugfix: set run_mode is not work

TingquanGao · TingquanGao · commit 46adcda713fc · 2025-06-24T12:36:51.000+08:00
diff --git a/paddlex/inference/utils/pp_option.py b/paddlex/inference/utils/pp_option.py
@@ -69,6 +69,7 @@ class PaddlePredictorOption(object):
 
     def __init__(self, model_name=None, **kwargs):
         super().__init__()
+        self._is_default_run_mode = True
         self._model_name = model_name
         self._cfg = {}
         self._init_option(**kwargs)
@@ -106,6 +107,10 @@ def _init_option(self, **kwargs):
                 raise Exception(
                     f"{k} is not supported to set! The supported option is: {self._get_settable_attributes()}"
                 )
+
+        if "run_mode" in self._cfg:
+            self._is_default_run_mode = False
+
         for k, v in self._get_default_config().items():
             self._cfg.setdefault(k, v)
 
@@ -153,9 +158,10 @@ def _update(self, k, v):
         self.changed = True
 
     def reset_run_mode_by_default(self, model_name=None, device_type=None):
-        model_name = model_name or self.model_name
-        device_type = device_type or self.device_type
-        self._update("run_mode", get_default_run_mode(model_name, device_type))
+        if self._is_default_run_mode:
+            model_name = model_name or self.model_name
+            device_type = device_type or self.device_type
+            self._update("run_mode", get_default_run_mode(model_name, device_type))
 
     @property
     def run_mode(self):
@@ -198,6 +204,7 @@ def run_mode(self, run_mode: str):
                 )
                 run_mode = "paddle"
 
+        self._is_default_run_mode = False
         self._update("run_mode", run_mode)
 
     @property
