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 b452f1e9156bacf734241aebbf149079ac3fcf9b..abfd74df6306f665e44ea7a0f23fdb23d8c05c7a 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 187cc6963aee51fd7f4c1007a9058b783db248a6..c29a2d16f02455c39c6b2bc884cdb75571c9928e 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 3b8d8724b9313d54164185a25ee0a8fed6f09db9..5a2f623e2ab7fb1c5eee28fb347fcb442b5a905b 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 0297fa76c83b240bcb5fb302e8beb0d76111e65f..f5cf6710c2ed944a0165aa0df773e1efdee28ff4 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 d9b86c642be67925e7c89522f7170996161c2ff6..65ffc017c5cf7fe381333dd01eaf2c48f8286eb4 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