fix

paddlex/inference/models_new/base/predictor/base_predictor.py

@@ -66,7 +66,9 @@ class BasePredictor(ABC):
         super().__init__()
         self.model_dir = Path(model_dir)
         self.config = config if config else self.load_config(self.model_dir)
-
+        self.batch_sampler = self._build_batch_sampler()
+        self.result_class = self._get_result_class()
+        
         # alias predict() to the __call__()
         self.predict = self.__call__
         self.benchmark = None

paddlex/inference/models_new/base/predictor/basic_predictor.py

@@ -55,8 +55,7 @@ class BasicPredictor(
         if device:
             pp_option.device = device
         self.pp_option = pp_option
-        self.batch_sampler = self._build_batch_sampler()
-        self.result_class = self._get_result_class()
+
         logging.debug(f"{self.__class__.__name__}: {self.model_dir}")
         self.benchmark = benchmark