From 707d0769a53718ffa65c041854484762284b4c1b Mon Sep 17 00:00:00 2001
From: stephon <lubin10@baidu.com>
Date: Tue, 9 Nov 2021 11:17:56 +0000
Subject: [PATCH] update muti_node log name

---
 test_tipc/test_train_inference_python.sh | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index eaeaf968..0bb73690 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -303,8 +303,16 @@ else
                 set_batchsize=$(func_set_params "${train_batch_key}" "${train_batch_value}")
                 set_train_params1=$(func_set_params "${train_param_key1}" "${train_param_value1}")
                 set_use_gpu=$(func_set_params "${train_use_gpu_key}" "${use_gpu}")
-                save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}"
-                
+                if [ ${#ips} -le 26 ];then
+                    save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}"
+                else
+                    IFS=","
+                    ips_array=(${ips})
+                    IFS="|"
+                    nodes=${#ips_array[@]}
+                    save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}"
+                fi
+
                 # load pretrain from norm training if current trainer is pact or fpgm trainer
                 if [ ${trainer} = ${pact_key} ] || [ ${trainer} = ${fpgm_key} ]; then
                     set_pretrain="${load_norm_train_model}"
-- 
GitLab