From 22c93f5f0aa002dcaa4505a2103388db7ee8a1aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E5=85=B4=E6=88=90?= <1358493914@qq.com>
Date: Thu, 1 Jun 2023 10:59:13 +0000
Subject: [PATCH] update
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 高兴成 <1358493914@qq.com>
---
 .../test/train_performance_bs256_hw192_16p.sh               | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/test/train_performance_bs256_hw192_16p.sh b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/test/train_performance_bs256_hw192_16p.sh
index d49691de8..e267c4fdb 100644
--- a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/test/train_performance_bs256_hw192_16p.sh
+++ b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/test/train_performance_bs256_hw192_16p.sh
@@ -107,6 +107,12 @@ if [[ ${fp32} == "--hf32" ]];then
   export ENABLE_HF32_EXECUTION=1
 fi
 
+if [[ $conf_path == "" ]];then
+    fix_node_ip=$fix_node_ip
+    one_node_ip=$one_node_ip
+else
+    one_node_ip=`find $conf_path -name "server_*_0.info"|awk -F "server_" '{print $2}'|awk -F "_" '{print $1}'`
+fi
 #新增适配集群环境变量
 export CM_CHIEF_IP=${one_node_ip}   #主节点ip，所有服务器一致
 export CM_CHIEF_PORT=29688          #通信端口，所有服务器一致
-- 
Gitee