From f538a55bf2ba42564366b5dae6c7b2cfe252f982 Mon Sep 17 00:00:00 2001 From: dongshuilong 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