diff --git a/test/params.txt b/test/params.txt index 9a4b52e880a39e8466747177d82c3b562cbcd172..67b2f8f18b09304f68eb9a902225ad6bb6f61436 100644 --- a/test/params.txt +++ b/test/params.txt @@ -1,6 +1,6 @@ train_model_list: ocr_det gpu_list: -1|0|0,1 -auto_cast_list: False|True +auto_cast_list: False trainer_list: norm|quant|prune python: python3.7 @@ -11,7 +11,6 @@ cpu_threads_list: 1|6 rec_batch_size_list: 1|6 gpu_trt_list: True|False gpu_precision_list: fp32|fp16|int8 -img_dir: /paddle/OCR/test_set/benchmark_eval epoch: 10 checkpoints: None diff --git a/test/test.sh b/test/test.sh index 6ba13c195ae3c5722b30fd7988aea3ad9545a599..1bc5a8f60cc652ac0fa266c3552f14abe7af6c41 100644 --- a/test/test.sh +++ b/test/test.sh @@ -50,11 +50,10 @@ cpu_threads_list=$(func_parser "${lines[8]}") rec_batch_size_list=$(func_parser "${lines[9]}") gpu_trt_list=$(func_parser "${lines[10]}") gpu_precision_list=$(func_parser "${lines[11]}") -img_dir=$(func_parser "${lines[12]}") - +img_dir="./train_data/icdar2015/text_localization/ch4_test_images/" # train superparameters -epoch=$(func_parser "${lines[13]}") -checkpoints=$(func_parser "${lines[14]}") +epoch=$(func_parser "${lines[12]}") +checkpoints=$(func_parser "${lines[13]}") for train_model in ${train_model_list[*]}; do