fix set epoch and batchsize

tests/ocr_det_params.txt

@@ -4,9 +4,9 @@ python:python3.7
 gpu_list:0|0,1
 Global.use_gpu:True|True
 Global.auto_cast:False
-Global.epoch_num:2
+Global.epoch_num:lite_train_infer=2|whole_train_infer=300
 Global.save_model_dir:./output/
-Train.loader.batch_size_per_card:2
+Train.loader.batch_size_per_card:lite_train_infer=2|whole_train_infer=4
 Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./train_data/icdar2015/text_localization/ch4_test_images/

tests/test.sh

@@ -34,6 +34,28 @@ function func_set_params(){
         echo "${key}=${value}"
     fi
 }
+function func_parser_params(){
+    strs=$1
+    IFS=":"
+    array=(${strs})
+    key=${array[0]}
+    tmp=${array[1]}
+    IFS="|"
+    res=""
+    for _params in ${tmp[*]}; do
+        IFS="="
+        array=(${_params})
+        mode=${array[0]}
+        value=${array[1]}
+        if [[ ${mode} = ${MODE} ]]; then
+            IFS="|"
+            echo $(func_set_params "${mode}" "${value}")
+            break
+        fi
+        IFS="|"
+    done
+    echo ${res}
+}
 function status_check(){
     last_status=$1   # the exit code
     run_command=$2
@@ -55,10 +77,10 @@ train_use_gpu_value=$(func_parser_value "${lines[4]}")
 autocast_list=$(func_parser_value "${lines[5]}")
 autocast_key=$(func_parser_key "${lines[5]}")
 epoch_key=$(func_parser_key "${lines[6]}")
-epoch_num=$(func_parser_value "${lines[6]}")
+epoch_num=$(func_parser_params "${lines[6]}")
 save_model_key=$(func_parser_key "${lines[7]}")
 train_batch_key=$(func_parser_key "${lines[8]}")
-train_batch_value=$(func_parser_value "${lines[8]}")
+train_batch_value=$(func_parser_params "${lines[8]}")
 pretrain_model_key=$(func_parser_key "${lines[9]}")
 pretrain_model_value=$(func_parser_value "${lines[9]}")
 train_model_name=$(func_parser_value "${lines[10]}")