From 0dfbadfb3f1dfe47dfd9c4cde98069f75f2540dd Mon Sep 17 00:00:00 2001 From: cnn Date: Fri, 21 May 2021 19:14:49 +0800 Subject: [PATCH] fix doc, set min_subgraph_size of s2anet (#3113) --- deploy/python/README.md | 2 +- ppdet/engine/export_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deploy/python/README.md b/deploy/python/README.md index 43622a363..3f4cd5832 100644 --- a/deploy/python/README.md +++ b/deploy/python/README.md @@ -39,7 +39,7 @@ python deploy/python/infer.py --model_dir=./inference/yolov3_mobilenet_v1_roadsi | --threshold | No|预测得分的阈值,默认为0.5| | --output_dir | No|可视化结果保存的根目录,默认为output/| | --run_benchmark | No| 是否运行benchmark,同时需指定`--image_file`或`--image_dir` | -| --use_mkldnn | No | CPU预测中是否开启MKLDNN加速 | +| --enable_mkldnn | No | CPU预测中是否开启MKLDNN加速 | | --cpu_threads | No| 设置cpu线程数,默认为1 | 说明: diff --git a/ppdet/engine/export_utils.py b/ppdet/engine/export_utils.py index 0bff32cdb..04506dc9d 100644 --- a/ppdet/engine/export_utils.py +++ b/ppdet/engine/export_utils.py @@ -31,7 +31,7 @@ TRT_MIN_SUBGRAPH = { 'SSD': 60, 'RCNN': 40, 'RetinaNet': 40, - 'S2ANet': 40, + 'S2ANet': 80, 'EfficientDet': 40, 'Face': 3, 'TTFNet': 60, -- GitLab