diff --git a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt index db710d91ff..e8d3cf2f27 100644 --- a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt +++ b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt @@ -1,8 +1,8 @@ ===========================train_params=========================== -model_name:det_mv3_db_v2.0_benchmark +model_name:det_mv3_db_v2.0 python:python -gpu_list:0|0,1 -Global.use_gpu:True|True +gpu_list:0 +Global.use_gpu:True Global.auto_cast:null Global.epoch_num:benchmark_train=2 Global.save_model_dir:./output/ diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 2753aef4b3..ef0272e0d5 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -22,7 +22,7 @@ trainer_list=$(func_parser_value "${lines[14]}") if [ ${MODE} = "benchmark_train" ];then pip install -r requirements.txt - if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then + if [[ ${model_name} =~ "det_mv3_db_v2.0" ]];then rm -rf ./train_data/icdar2015 wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate