From 4066478ea4a297312570c078de41191b5982ca6b Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Thu, 8 Jul 2021 11:56:52 +0800
Subject: [PATCH] fix precision

---
 tools/infer/utility.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index e464722f..bf05a0db 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -164,7 +164,7 @@ def create_predictor(args, mode, logger):
         config.enable_use_gpu(args.gpu_mem, 0)
         if args.use_tensorrt:
             config.enable_tensorrt_engine(
-                precision_mode=inference.PrecisionType.Float32,
+                precision_mode=precision,
                 max_batch_size=args.max_batch_size,
                 min_subgraph_size=args.min_subgraph_size)
             # skip the minmum trt subgraph
-- 
GitLab