From 8dae8b5e6189b83210f601851ace3fb8c94859fb Mon Sep 17 00:00:00 2001 From: qingqing01 Date: Mon, 10 Feb 2020 18:02:45 +0800 Subject: [PATCH] Change min_subgraph_size to 40 in tools/cpp_demo.yml (#222) --- docs/advanced_tutorials/inference/INFERENCE.md | 2 +- tools/cpp_demo.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/advanced_tutorials/inference/INFERENCE.md b/docs/advanced_tutorials/inference/INFERENCE.md index 0baceb6c0..b15f6c9c0 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 0940c1a84..96f111676 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 -- GitLab