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 b86b0814f4eef4df0f394a2836f2e217408ea8e1..eee7293b13adaa4cc537a8e1c25c8dba878f1041 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,7 @@ 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
+   
   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}
@@ -118,7 +116,8 @@ done
 
 wait
 sed -i 's///g' ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log
-sed -i 's/
/\n/g' ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log
+sed -i 's/
+/\n/g' ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log
 ##################获取训练数据################
 #训练结束时间，不需要修改
 end_time=$(date +%s)