diff --git a/deploy/python/README.md b/deploy/python/README.md index 43622a363fc79d9248b5f452bf35eaf05eb2263c..3f4cd5832f6142909ee755a3a7edd3767ef79fcd 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 0bff32cdbd927e3c65c3acdd35f491b122b7b34a..04506dc9d0bee9eebf356984ec7692d26ec4b32a 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,