diff --git a/docs/advanced_tutorials/inference/INFERENCE.md b/docs/advanced_tutorials/inference/INFERENCE.md index 0baceb6c03d50826c0d10d5d16a2e58b7a913a00..b15f6c9c0fb876a0c9aa43128793d41eed33ab6b 100644 --- a/docs/advanced_tutorials/inference/INFERENCE.md +++ b/docs/advanced_tutorials/inference/INFERENCE.md @@ -23,7 +23,7 @@ python tools/cpp_infer.py --model_path=inference_model/faster_rcnn_r50_1x/ --con **注意** 1. 设置shape时必须保持与模型导出时shape大小一致; -2. `min_subgraph_size`的设置与模型arch相关,对部分arch需要调大该参数,一般设置为40适用于所有模型。适当的调小`min_subgraph_size`会对预测有小幅加速效果,例如YOLO中该参数可设置为3。 +2. `min_subgraph_size`的设置与模型arch相关,对部分arch需要调大该参数,一般设置为40适用于所有模型。适当的调小`min_subgraph_size`会对预测有加速效果,例如YOLO中该参数可设置为3。 ## Paddle环境搭建 diff --git a/tools/cpp_demo.yml b/tools/cpp_demo.yml index 0940c1a84545da8e582f203675992d722809cf93..96f111676d9339676245c662b7c17e34f387e5da 100644 --- a/tools/cpp_demo.yml +++ b/tools/cpp_demo.yml @@ -2,7 +2,7 @@ mode: trt_fp32 # trt_fp32, trt_fp16, trt_int8, fluid arch: RCNN # YOLO, SSD, RCNN, RetinaNet -min_subgraph_size: 20 # need 3 for YOLO arch +min_subgraph_size: 40 # need 3 for YOLO arch use_python_inference: False # whether to use python inference # visualize the predicted image