diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 1d89b3c4da377ec25b45a9a3de6a28e43d6fc228..8a701d359227e532414b726ff8e84a37af3adf59 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -154,7 +154,7 @@ class Predictor: predictor_config = fluid.core.NativeConfig() predictor_config.prog_file = self.config.model_file predictor_config.param_file = self.config.param_file - predictor_config.use_gpu = config.use_gpu + predictor_config.use_gpu = self.config.use_gpu predictor_config.device = 0 predictor_config.fraction_of_gpu_memory = 0 elif self.config.predictor_mode == "ANALYSIS":