fix precision

4066478e · Double_V · GitHub · 7fd27329 · 4066478e
Unverified Commit 4066478e authored Jul 08, 2021 by Double_V Committed by GitHub Jul 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/infer/utility.py tools/infer/utility.py +1 -1

No files found.
--- 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