From 83149576de2ab6b664c8608ed0106bd5967b8cc2 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 8 Jun 2021 16:18:08 +0800 Subject: [PATCH] remove img_dir from params.txt --- test/params.txt | 3 +-- test/test.sh | 7 +++---- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/test/params.txt b/test/params.txt index 9a4b52e8..67b2f8f1 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 6ba13c19..1bc5a8f6 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 -- GitLab