diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index e76c60a5367138c26c40b5c5d8603ae8550e7edf..da8cb37e022016490188165c7dad648522fcc556 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -92,7 +92,7 @@ benchmark_value=$(func_parser_value "${lines[49]}")
 infer_key1=$(func_parser_key "${lines[50]}")
 infer_value1=$(func_parser_value "${lines[50]}")
 
-LOG_PATH="./test_tipc/output"
+LOG_PATH="./test_tipc/output/${model_name}"
 mkdir -p ${LOG_PATH}
 status_log="${LOG_PATH}/results_python.log"
 
@@ -305,7 +305,7 @@ else
                     status_check $? "${export_cmd}" "${status_log}"
 
                     #run inference
-                    save_export_model_dir="${save_export_value}/${model_name}"
+                    save_export_model_dir="${save_log}/${model_name}"
                     eval $env
                     func_inference "${python}" "${inference_py}" "${save_export_model_dir}" "${LOG_PATH}" "${train_infer_img_dir}" "${flag_quant}"