From f538a55bf2ba42564366b5dae6c7b2cfe252f982 Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Wed, 6 Jul 2022 07:06:07 +0000
Subject: [PATCH] add ResNet50 log number in benchmark

---
 test_tipc/static/ResNet50/benchmark_common/prepare.sh       | 4 +++-
 test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/test_tipc/static/ResNet50/benchmark_common/prepare.sh b/test_tipc/static/ResNet50/benchmark_common/prepare.sh
index bdf34737f..1dc3b7a47 100644
--- a/test_tipc/static/ResNet50/benchmark_common/prepare.sh
+++ b/test_tipc/static/ResNet50/benchmark_common/prepare.sh
@@ -7,5 +7,7 @@ wget -nc https://paddle-imagenet-models-name.bj.bcebos.com/data/ImageNet1k/ILSVR
 tar xf ILSVRC2012_val.tar
 ln -s ILSVRC2012_val ILSVRC2012
 cd ILSVRC2012
-ln -s val_list.txt  train_list.txt
+for ((i=1; i<=4; i++));do
+  cat val_list.txt >> train_list.txt
+done
 cd ../../
diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
index c84ce46be..5146e6ccf 100644
--- a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
+++ b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
@@ -63,7 +63,7 @@ function _train(){
     esac
 
     echo "train_cmd: ${train_cmd}  log_file: ${log_file}"
-    timeout 5m ${train_cmd} > ${log_file} 2>&1
+    timeout 10m ${train_cmd} > ${log_file} 2>&1
     if [ $? -ne 0 ];then
         echo -e "${model_name}, FAIL"
     else