Skip to content

P
PaddleOCR

s920243400 / PaddleOCR
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleOCR

通知 1
Star 1
Fork 0
P
PaddleOCR
未验证 提交 96eb6287 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者： GitHub
浏览文件
操作

fix trt int8 inference (#6387)

上级 f2c0a1a2
显示空白变更内容
内联 并排
Showing with 2 addition and 1 deletion
tools/infer/utility.py
浏览文件 @ 96eb6287
...@@ -202,7 +202,8 @@ def create_predictor(args, mode, logger): ...@@ -202,7 +202,8 @@ def create_predictor(args, mode, logger):
workspace_size=1 << 30, workspace_size=1 << 30,
precision_mode=precision, 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,
use_calib_mode=False)
# skip the minmum trt subgraph # skip the minmum trt subgraph
use_dynamic_shape = True use_dynamic_shape = True
if mode == "det": if mode == "det":
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或