diff --git a/tests/test_tipc/benchmark_train.sh b/tests/test_tipc/benchmark_train.sh index 7f0382ac5..7b24cfcbe 100644 --- 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}