adjust the min_subgraph_size for trt
@@ -235,7 +235,7 @@ def create_predictor(args, mode, logger):
config.enable_tensorrt_engine(
precision_mode=inference.PrecisionType.Float32,
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:
min_input_shape = {
"x": [1, 3, 50, 50],
