fix precision

2021-07-08 11:56:52 +08:00 · 2021-07-08 11:56:52 +08:00 · 4066478ea4
parent 7fd27329ef
commit 4066478ea4
1 changed files with 1 additions and 1 deletions
--- 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