From 0c3975b9945ac3e90fee1deaa4337b7d891f6e8b Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Fri, 26 Nov 2021 15:07:01 +0800 Subject: [PATCH] [TIPC] fix cascade rcnn configs, test=document_fix (#4727) --- .../cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 ++-- ..._model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt | 4 ++-- ...e_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt | 4 ++-- .../cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 ++-- ...ascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt | 4 ++-- 5 files changed, 10 insertions(+), 10 deletions(-) diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt index b452f1e91..abfd74df6 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 187cc6963..c29a2d16f 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -20,10 +20,10 @@ inference:./deploy/cpp/build/main --device:gpu|cpu --use_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index 3b8d8724b..5a2f623e2 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt index 0297fa76c..f5cf6710c 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index d9b86c642..65ffc017c 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file -- GitLab