From f326bd8b4c5afdff9c5563981c498f51eef0e657 Mon Sep 17 00:00:00 2001
From: Zhao Yuting <91456992+THUzyt21@users.noreply.github.com>
Date: Tue, 27 Sep 2022 16:55:28 +0800
Subject: [PATCH] Update test_train_inference_python.sh

fix a small bug
---
 tests/test_tipc/test_train_inference_python.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_tipc/test_train_inference_python.sh b/tests/test_tipc/test_train_inference_python.sh
index ef5747b44..04ecf8c98 100644
--- a/tests/test_tipc/test_train_inference_python.sh
+++ b/tests/test_tipc/test_train_inference_python.sh
@@ -20,10 +20,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_params "${lines[6]}" "${MODE}")
+epoch_num=`echo  "${lines[6]}" | tr -cd "[0-9]" `
 save_model_key=$(func_parser_key "${lines[7]}")
 train_batch_key=$(func_parser_key "${lines[8]}")
-train_batch_value=$(func_parser_params "${lines[8]}" "${MODE}")
+train_batch_value=`echo  "${lines[8]}" | tr -cd "[0-9]" `
 pretrain_model_key=$(func_parser_key "${lines[9]}")
 pretrain_model_value=$(func_parser_value "${lines[9]}")
 train_model_name=$(func_parser_value "${lines[10]}")
@@ -374,4 +374,4 @@ else
             done  # done with:    for trainer in ${trainer_list[*]}; do 
         done      # done with:    for autocast in ${autocast_list[*]}; do 
     done          # done with:    for gpu in ${gpu_list[*]}; do
-fi  # end if [ ${MODE} = "infer" ]; then
\ No newline at end of file
+fi  # end if [ ${MODE} = "infer" ]; then