Skip to content

GitLab

"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "86b90aa93566ae94e9d7b4fd9b3677e998c68c5c"
Commit 796898e0 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

adjust the min_subgraph_size for trt

parent 7ad66311
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/infer/utility.py
View file @ 796898e0
...@@ -235,7 +235,7 @@ def create_predictor(args, mode, logger): ...@@ -235,7 +235,7 @@ def create_predictor(args, mode, logger):
config.enable_tensorrt_engine( config.enable_tensorrt_engine(
precision_mode=inference.PrecisionType.Float32, precision_mode=inference.PrecisionType.Float32,
max_batch_size=args.max_batch_size, max_batch_size=args.max_batch_size,
min_subgraph_size=3) # skip the minmum trt subgraph min_subgraph_size=10) # skip the minmum trt subgraph
if mode == "det" and "mobile" in model_file_path: if mode == "det" and "mobile" in model_file_path:
min_input_shape = { min_input_shape = {
"x": [1, 3, 50, 50], "x": [1, 3, 50, 50],
......
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