From 2b793cd8348f2f92a869737cbaa8a267dbcc82a5 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 8 May 2020 18:33:00 +0800
Subject: [PATCH] fix deploy/python/infer.py invaild symbol (#616)

---
 deploy/python/infer.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 252f3e2b4..9c544e8b8 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -306,16 +306,16 @@ def load_predictor(model_dir,
                    batch_size=1,
                    use_gpu=False,
                    min_subgraph_size=3):
-    """set AnalysisConfig，generate AnalysisPredictor
+    """set AnalysisConfig, generate AnalysisPredictor
     Args:
         model_dir (str): root path of __model__ and __params__
         use_gpu (bool): whether use gpu
     Returns:
         predictor (PaddlePredictor): AnalysisPredictor
     Raises:
-        ValueError: predict by TensorRT need use_gpu == True 
+        ValueError: predict by TensorRT need use_gpu == True.
     """
-    if not use_gpu and run_mode == 'fluid':
+    if not use_gpu and not run_mode == 'fluid':
         raise ValueError(
             "Predict by TensorRT mode: {}, expect use_gpu==True, but use_gpu == {}"
             .format(run_mode, use_gpu))
-- 
GitLab