diff --git a/test/params.txt b/test/params.txt index 8101714f2e85259fc225318d1ae42d44faad286b..9a4b52e880a39e8466747177d82c3b562cbcd172 100644 --- a/test/params.txt +++ b/test/params.txt @@ -1,4 +1,4 @@ -train_model_list: det;benchmark/benchmark_det.yml +train_model_list: ocr_det gpu_list: -1|0|0,1 auto_cast_list: False|True trainer_list: norm|quant|prune diff --git a/test/test.sh b/test/test.sh index 559dc4b02ab4fcf714f4d8aadd1e0c027cc19972..a17d3274a1bca2a4bda50def4f2e327da16614e0 100644 --- a/test/test.sh +++ b/test/test.sh @@ -55,10 +55,10 @@ checkpoints=$(func_parser "${lines[14]}") for train_model in ${train_model_list[*]}; do - if [ ${train_model} = "det" ];then + if [ ${train_model} = "ocr_det" ];then model_name="det" yml_file="configs/det/det_mv3_db.yml" - elif [ ${train_model} = "rec" ];then + elif [ ${train_model} = "ocr_rec" ];then model_name="rec" yml_file="configs/rec/rec_mv3_none_bilstm_ctc.yml" else