Browse Source

!10878 simplepose-shell

From: @wukesong
Reviewed-by: @wuxuejian,@c_34
Signed-off-by: @wuxuejian
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
eb147d0911
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      model_zoo/official/cv/simple_pose/scripts/train_distributed.sh

+ 1
- 2
model_zoo/official/cv/simple_pose/scripts/train_distributed.sh View File

@@ -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


Loading…
Cancel
Save