Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "c7a50dbdc290516a1dcfc0d9d8a9831550ffe32b"
Unverified Commit 02b75a50 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #3293 from PaddlePaddle/LDOUBLEV-patch-1 

fix precision
parents 7fd27329 4066478e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/utility.py
View file @ 02b75a50
...@@ -164,7 +164,7 @@ def create_predictor(args, mode, logger): ...@@ -164,7 +164,7 @@ def create_predictor(args, mode, logger):
config.enable_use_gpu(args.gpu_mem, 0) config.enable_use_gpu(args.gpu_mem, 0)
if args.use_tensorrt: if args.use_tensorrt:
config.enable_tensorrt_engine( config.enable_tensorrt_engine(
precision_mode=inference.PrecisionType.Float32, precision_mode=precision,
max_batch_size=args.max_batch_size, max_batch_size=args.max_batch_size,
min_subgraph_size=args.min_subgraph_size) min_subgraph_size=args.min_subgraph_size)
# skip the minmum trt subgraph # skip the minmum trt subgraph
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment