From 8e5bde6c39e32593abccdcde8286c48978ecfa3b Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Tue, 1 Dec 2020 01:20:10 +0800
Subject: [PATCH] Update predict.py

---
 tools/infer/predict.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/infer/predict.py b/tools/infer/predict.py
index 849d4064..3cdff940 100644
--- a/tools/infer/predict.py
+++ b/tools/infer/predict.py
@@ -39,8 +39,8 @@ def create_paddle_predictor(args):
     config.switch_ir_optim(args.ir_optim)  # default true
     if args.use_tensorrt:
         config.enable_tensorrt_engine(
-            precision_mode=Config.PrecisionType.Half
-            if args.use_fp16 else Config.PrecisionType.Float32,
+            precision_mode=Config.Precision.Half
+            if args.use_fp16 else Config.Precision.Float32,
             max_batch_size=args.batch_size)
 
     config.enable_memory_optim()
-- 
GitLab