diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index 4ad83c66977540d73bdc9bedb8b93bf465e8b6fc..16024caf485b737e6777b6068c6c005dfc16d575 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -316,14 +316,14 @@ else
                 set_train_params1=$(func_set_params "${train_param_key1}" "${train_param_value1}")
                 set_use_gpu=$(func_set_params "${train_use_gpu_key}" "${train_use_gpu}")
                 if [ ${#ips} -le 26 ];then
-                    save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}"
+                    save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}"
                     nodes=1
                 else
                     IFS=","
                     ips_array=(${ips})
                     IFS="|"
                     nodes=${#ips_array[@]}
-                    save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}"
+                    save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}"
                 fi