!6245 modify the way of read device num

Merge pull request !6245 from hwjiaorui/master
2020-09-15 16:27:04 +08:00 · 2020-09-15 16:27:04 +08:00 · 73d4ceeaf7
parent ff42cd87b2 1aa8e78491
commit 73d4ceeaf7
1 changed files with 4 additions and 3 deletions
--- a/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh
+++ b/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh
@ -91,11 +91,12 @@ run_ascend(){
    #device_num=${#device_id_list}
    cat $2 | awk -F "[device_id]" '/device_id/{print$0}' >temp.log
    array=$(cat temp.log | awk -F "[:]" '/device_id/{print$2}')
-    rm temp.log
+    
    IFS=" " read -ra device_list <<<$array
    first_device=${device_list[0]:1:1}
-    device_num=${#device_list[*]}
-
+    #device_num=${#device_list[*]}
+    device_num=$(cat temp.log | wc -l)
+    rm temp.log
    ulimit -u unlimited
    export DEVICE_NUM=${device_num}
    export RANK_SIZE=${device_num}