提交 ab0b1ace 编写于 作者: L LDOUBLEV

add params min_subgraph_size

上级 96a53fb4
...@@ -236,8 +236,8 @@ def create_predictor(args, mode, logger): ...@@ -236,8 +236,8 @@ 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=args. min_subgraph_size=args.min_subgraph_size)
min_subgraph_size) # skip the minmum trt subgraph # skip the minmum trt subgraph
if mode == "det": if mode == "det":
min_input_shape = { min_input_shape = {
"x": [1, 3, 50, 50], "x": [1, 3, 50, 50],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册