@@ -69,6 +69,7 @@ class PaddlePredictorOption(object):
6969
7070 def __init__ (self , model_name = None , ** kwargs ):
7171 super ().__init__ ()
72+ self ._is_default_run_mode = True
7273 self ._model_name = model_name
7374 self ._cfg = {}
7475 self ._init_option (** kwargs )
@@ -106,6 +107,10 @@ def _init_option(self, **kwargs):
106107 raise Exception (
107108 f"{ k } is not supported to set! The supported option is: { self ._get_settable_attributes ()} "
108109 )
110+
111+ if "run_mode" in self ._cfg :
112+ self ._is_default_run_mode = False
113+
109114 for k , v in self ._get_default_config ().items ():
110115 self ._cfg .setdefault (k , v )
111116
@@ -153,9 +158,10 @@ def _update(self, k, v):
153158 self .changed = True
154159
155160 def reset_run_mode_by_default (self , model_name = None , device_type = None ):
156- model_name = model_name or self .model_name
157- device_type = device_type or self .device_type
158- self ._update ("run_mode" , get_default_run_mode (model_name , device_type ))
161+ if self ._is_default_run_mode :
162+ model_name = model_name or self .model_name
163+ device_type = device_type or self .device_type
164+ self ._update ("run_mode" , get_default_run_mode (model_name , device_type ))
159165
160166 @property
161167 def run_mode (self ):
@@ -198,6 +204,7 @@ def run_mode(self, run_mode: str):
198204 )
199205 run_mode = "paddle"
200206
207+ self ._is_default_run_mode = False
201208 self ._update ("run_mode" , run_mode )
202209
203210 @property
0 commit comments