From c81d95a837c36d6c7b3b2d495a7b6450a193c2fe Mon Sep 17 00:00:00 2001
From: Zhao Yuting <91456992+THUzyt21@users.noreply.github.com>
Date: Thu, 15 Sep 2022 14:22:06 +0800
Subject: [PATCH] Update benchmark_train.sh

Use more gpu to speed up the training process.
---
 tests/test_tipc/benchmark_train.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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}