fix benchmark model_repo

4 years ago · a91ee3d982
parent 41d194d123
commit a91ee3d982
1 changed files with 2 additions and 2 deletions
--- a/tests/test_tipc/benchmark_train.sh
+++ b/tests/test_tipc/benchmark_train.sh
@ -61,7 +61,7 @@ function get_repo_name(){
    cur_dir=$(pwd)
    IFS="/"
    arr=(${cur_dir})
-    echo ${arr[-1]}
+    echo ${arr[-2]}
 }

 FILENAME=$1
@ -250,4 +250,4 @@ for batch_size in ${batch_size_list[*]}; do
            fi
        done
    done
-done
+done