[TIPC] Solve bisenet,ocrnet_hrnetw18 overtime problem in lite_train_l…

…ite_infer mode (PaddlePaddle#1937)
PeiyuLau · Mar 29, 2022 · fc714ac · fc714ac
1 parent c51e0e1
commit fc714ac
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/test_tipc/configs/bisenetv2/train_infer_python.txt b/test_tipc/configs/bisenetv2/train_infer_python.txt
@@ -9,7 +9,7 @@ Global.use_gpu:null|null
 --batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=3
 --model_path:null
 train_model_name:best_model/model.pdparams
-train_infer_img_dir:test_tipc/data/cityscapes/val.list
+train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list
 null:null
 ##
 trainer:norm
@@ -45,7 +45,7 @@ inference:deploy/python/infer.py --save_dir ./test_tipc/output/bisenetv2/
 --use_trt:False
 --precision:fp32|int8|fp16
 --config:
---image_path:./test_tipc/data/cityscapes/val.list
+--image_path:./test_tipc/data/cityscapes/cityscapes_val_5.list
 --save_log_path:null
 --benchmark:True
 --save_dir:

diff --git a/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt b/test_tipc/configs/ocrnet_hrnetw18/train_infer_python.txt
@@ -9,7 +9,7 @@ Global.use_gpu:null|null
 --batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=3
 --model_path:null
 train_model_name:best_model/model.pdparams
-train_infer_img_dir:test_tipc/data/cityscapes/val.list
+train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list
 null:null
 ##
 trainer:norm
@@ -45,7 +45,7 @@ inference:deploy/python/infer.py
 --use_trt:False
 --precision:fp32|int8|fp16
 --config:
---image_path:./test_tipc/data/cityscapes/val.list
+--image_path:./test_tipc/data/cityscapes/cityscapes_val_5.list
 --save_log_path:null
 --benchmark:True
 --save_dir:

diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
@@ -76,6 +76,6 @@ else
     fi
 fi
 
-if [ ${model_name} == "enet" ];then
+if [ ${model_name} == "enet" ] || [ ${model_name} == "bisenetv2" ] || [ ${model_name} == "ocrnet_hrnetw18" ];then
     cp ./test_tipc/data/cityscapes_val_5.list ./test_tipc/data/cityscapes
 fi