From ea2fb83fa89a9c742e3813e6f946538befb0e5d9 Mon Sep 17 00:00:00 2001
From: limingxing517 <limingxing517@163.com>
Date: Fri, 12 May 2023 15:26:34 +0800
Subject: [PATCH] update

---
 .../test/train_performance_1p.sh                         | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/TensorFlow/built-in/recommendation/DIN_ID0190_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/recommendation/DIN_ID0190_for_TensorFlow/test/train_performance_1p.sh
index b86b0814f..57afc90d5 100644
--- a/TensorFlow/built-in/recommendation/DIN_ID0190_for_TensorFlow/test/train_performance_1p.sh
+++ b/TensorFlow/built-in/recommendation/DIN_ID0190_for_TensorFlow/test/train_performance_1p.sh
@@ -87,6 +87,7 @@ fi
 cd $cur_path/examples
 sed -i  "s|./data|$data_path|g" din_demo.py
 sed -i  "s|epochs=5|epochs=2|g" din_demo.py
+sed -i  "s|rank_id =|#rank_id =|g" din_demo.py
 
 
 RANK_ID_START=0
@@ -94,10 +95,10 @@ RANK_SIZE=1
 
 for((RANK_ID=$RANK_ID_START;RANK_ID<$((RANK_SIZE+RANK_ID_START));RANK_ID++));
 do
-   echo "Device ID: $RANK_ID"
-   export RANK_ID=$RANK_ID
-   export ASCEND_DEVICE_ID=$RANK_ID
-   ASCEND_DEVICE_ID=$RANK_ID
+   #echo "Device ID: $RANK_ID"
+   #export RANK_ID=$RANK_ID
+   #export ASCEND_DEVICE_ID=$RANK_ID
+   #ASCEND_DEVICE_ID=$RANK_ID
   if [   -d $cur_path/test/output/${ASCEND_DEVICE_ID} ];then
      rm -rf $cur_path/test/output/${ASCEND_DEVICE_ID}
      mkdir -p $cur_path/test/output/${ASCEND_DEVICE_ID}
-- 
Gitee