diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index 7b7d590ab29b55c9a1a3bd6ae852f6af2163b482..88274731aabb60a55bbfd0d0103269fd523e2224 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -274,7 +274,11 @@ else # export FLAGS_cudnn_deterministic=True sleep 5 eval $cmd - eval "cat ${save_log}/${model_name}/train.log >> ${save_log}.log" + if [[ $FILENAME == *GeneralRecognition* ]]; then + eval "cat ${save_log}/RecModel/train.log >> ${save_log}.log" + else + eval "cat ${save_log}/${model_name}/train.log >> ${save_log}.log" + fi status_check $? "${cmd}" "${status_log}" "${model_name}" "${save_log}.log" sleep 5