Browse Source

modify simplepose shell

tags/v1.2.0-rc1
wukesong 5 years ago
parent
commit
6fa2102bd6
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