From d83de4ea3301333bd471295328ace4d29814d019 Mon Sep 17 00:00:00 2001
From: wukesong <wukesong1@huawei.com>
Date: Thu, 24 Dec 2020 20:21:57 +0800
Subject: [PATCH] modify tiny-darknet

---
 model_zoo/official/cv/tinydarknet/scripts/run_train.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model_zoo/official/cv/tinydarknet/scripts/run_train.sh b/model_zoo/official/cv/tinydarknet/scripts/run_train.sh
index 300ec840db..0601c02a59 100644
--- a/model_zoo/official/cv/tinydarknet/scripts/run_train.sh
+++ b/model_zoo/official/cv/tinydarknet/scripts/run_train.sh
@@ -50,7 +50,7 @@ echo "RANK_TABLE_FILE=${RANK_TABLE_FILE}"
 
 export SERVER_ID=0
 rank_start=$((DEVICE_NUM * SERVER_ID))
-for((i=0; i<${SERVER_NUM}; i++))
+for((i=0; i<${DEVICE_NUM}; i++))
 do
     export DEVICE_ID=$i
     export RANK_ID=$((rank_start + i))
@@ -59,7 +59,7 @@ do
     cp -r ../src ./train_parallel$i
     cp ../train.py ./train_parallel$i
     echo "start training for rank $RANK_ID, device $DEVICE_ID, $dataset_type"
-    cd .train_parallel$i || exit
+    cd ./train_parallel$i || exit
     env > env.log
     python train.py --device_id=$i --dataset_name=$dataset_type> log 2>&1 &
     cd ..