From 8a66e2d77ab5cc1209d574f6aba87a046aa70f93 Mon Sep 17 00:00:00 2001 From: cnn Date: Fri, 9 Jul 2021 16:05:04 +0800 Subject: [PATCH] add parameter description of trt_calib_mode, test=document_fix (#3644) --- deploy/python/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/deploy/python/README.md b/deploy/python/README.md index 131a5f112..fb4fd9a9a 100644 --- a/deploy/python/README.md +++ b/deploy/python/README.md @@ -41,6 +41,7 @@ python deploy/python/infer.py --model_dir=./inference/yolov3_mobilenet_v1_roadsi | --run_benchmark | Option| 是否运行benchmark，同时需指定`--image_file`或`--image_dir`，默认为False | | --enable_mkldnn | Option | CPU预测中是否开启MKLDNN加速，默认为False | | --cpu_threads | Option| 设置cpu线程数，默认为1 | +| --trt_calib_mode | Option| TensorRT是否使用校准功能，默认为False。使用TensorRT的int8功能时，需设置为True，使用PaddleSlim量化后的模型时需要设置为False | 说明： -- GitLab