未验证 提交 a26d964b 编写于 作者: B Bin Long 提交者: GitHub

fix deploy/infer.py typo

上级 4a54ed4d
......@@ -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":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册