Update benchmark_train.sh

Use more gpu to speed up the training process.
3 years ago · c81d95a837
parent 2a3bc9f65b
commit c81d95a837
1 changed files with 1 additions and 1 deletions
--- a/tests/test_tipc/benchmark_train.sh
+++ b/tests/test_tipc/benchmark_train.sh
@ -131,7 +131,7 @@ if  [ ! -n "$PARAMS" ] ;then
    IFS="|"
    batch_size_list=(${batch_size})
    fp_items_list=(${fp_items})
-    device_num_list=(N1C4)
+    device_num_list=(N1C8)
    run_mode="DP"
 else
    # parser params from input: modeltype_bs${bs_item}_${fp_item}_${run_mode}_${device_num}