提交 8dae8b5e 编写于 作者: Q qingqing01 提交者: GitHub

Change min_subgraph_size to 40 in tools/cpp_demo.yml (#222)

上级 4d096b4f
......@@ -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环境搭建
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册