mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

tipc: fix the keyword of ShiTu model 

In benchmark_train mode, the config name is benchmark.txt. So the keyword of ShiTu should be model_name instead of the config name.
pull/2314/head
gaotingquan 2022-09-15 13:37:51 +00:00 committed by Tingquan Gao
parent bfa1ac144a
commit 983d512a45
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test_tipc/test_train_inference_python.sh
View File

@ -274,7 +274,7 @@ else
# export FLAGS_cudnn_deterministic=True
sleep 5
eval $cmd
if [[ $FILENAME == *GeneralRecognition* ]]; then
if [[ $model_name == *GeneralRecognition* ]]; then
eval "cat ${save_log}/RecModel/train.log >> ${save_log}.log"
else
eval "cat ${save_log}/${model_name}/train.log >> ${save_log}.log"
@ -282,7 +282,7 @@ else
status_check $? "${cmd}" "${status_log}" "${model_name}" "${save_log}.log"
sleep 5
if [[ $FILENAME == *GeneralRecognition* ]]; then
if [[ $model_name == *GeneralRecognition* ]]; then
set_eval_pretrain=$(func_set_params "${pretrain_model_key}" "${save_log}/RecModel/${train_model_name}")
else
set_eval_pretrain=$(func_set_params "${pretrain_model_key}" "${save_log}/${model_name}/${train_model_name}")
@ -304,7 +304,7 @@ else
if [ ${run_export} != "null" ]; then
# run export model
save_infer_path="${save_log}"
if [[ $FILENAME == *GeneralRecognition* ]]; then
if [[ $model_name == *GeneralRecognition* ]]; then
set_eval_pretrain=$(func_set_params "${pretrain_model_key}" "${save_log}/RecModel/${train_model_name}")
else
set_export_weight=$(func_set_params "${export_weight}" "${save_log}/${model_name}/${train_model_name}")