From 6fa2102bd6402cad50b2c7140ae688e01b113624 Mon Sep 17 00:00:00 2001
From: wukesong <wukesong1@huawei.com>
Date: Thu, 31 Dec 2020 14:44:36 +0800
Subject: [PATCH] modify simplepose shell

---
 model_zoo/official/cv/simple_pose/scripts/train_distributed.sh | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/model_zoo/official/cv/simple_pose/scripts/train_distributed.sh b/model_zoo/official/cv/simple_pose/scripts/train_distributed.sh
index 3bf549b62f..882660df3b 100644
--- a/model_zoo/official/cv/simple_pose/scripts/train_distributed.sh
+++ b/model_zoo/official/cv/simple_pose/scripts/train_distributed.sh
@@ -20,10 +20,9 @@ echo "RANK_TABLE_FILE=$RANK_TABLE_FILE"
 export RANK_SIZE=$3
 SAVE_PATH=$2
 
-device=(0 1 2 3)
 for((i=0;i<RANK_SIZE;i++))
 do
-    export DEVICE_ID=${device[$i]}
+    export DEVICE_ID=$i
     export RANK_ID=$i
 
     rm -rf ./train_parallel$i