Skip to content

GitLab

Unverified Commit 9ffbacc4 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #6616 from Rm1n90/release/2.5 

Fix TensorRT failed due to some TRT inputs dynamic shape info not set
parents 8fa9cbd8 b70255af
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tools/infer/utility.py
View file @ 9ffbacc4
...@@ -276,6 +276,7 @@ def create_predictor(args, mode, logger): ...@@ -276,6 +276,7 @@ def create_predictor(args, mode, logger):
min_input_shape = {"x": [1, 3, imgH, 10]} min_input_shape = {"x": [1, 3, imgH, 10]}
max_input_shape = {"x": [args.rec_batch_num, 3, imgH, 2304]} max_input_shape = {"x": [args.rec_batch_num, 3, imgH, 2304]}
opt_input_shape = {"x": [args.rec_batch_num, 3, imgH, 320]} opt_input_shape = {"x": [args.rec_batch_num, 3, imgH, 320]}
config.exp_disable_tensorrt_ops(["transpose2"])
elif mode == "cls": elif mode == "cls":
min_input_shape = {"x": [1, 3, 48, 10]} min_input_shape = {"x": [1, 3, 48, 10]}
max_input_shape = {"x": [args.rec_batch_num, 3, 48, 1024]} max_input_shape = {"x": [args.rec_batch_num, 3, 48, 1024]}
......
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