yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-28 07:28:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix trt int8 inference (#6387)

Browse Source
This commit is contained in:
littletomatodonkey 2022-05-24 15:34:24 +08:00 committed by GitHub
parent f2c0a1a2fd
commit 96eb6287f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/infer/utility.py
View File

@ -202,7 +202,8 @@ def create_predictor(args, mode, logger):
workspace_size=1 << 30,
precision_mode=precision,
max_batch_size=args.max_batch_size,
min_subgraph_size=args.min_subgraph_size)
min_subgraph_size=args.min_subgraph_size,
use_calib_mode=False)
# skip the minmum trt subgraph
use_dynamic_shape = True
if mode == "det":