diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh index 5c4d4112ad691569914ccf9b84480db9b76fa024..bbec7c62aab656b2b1076ad478ecf6094b20e9b6 100644 --- a/test_tipc/benchmark_train.sh +++ b/test_tipc/benchmark_train.sh @@ -179,6 +179,22 @@ for batch_size in ${batch_size_list[*]}; do func_sed_params "$FILENAME" "${line_epoch}" "$epoch" gpu_id=$(set_gpu_id $device_num) + # if bs is big, then copy train_list.txt to generate more train log + # There are 5w image in train_list. And the train log printed interval is 10 iteration. + # At least 25 log number would be good to calculate ips for benchmark system. + # So the copy number for train_list is as follows: + total_batch_size=`echo $[$batch_size*${device_num:1:1}*${device_num:3:3}]` + copy_num=`echo $[$total_batch_size/200]` + if [ $copy_num -gt 1 ];then + cd dataset/ILSVRC2012 + rm -rf train_list.txt + for ((i=1; i <=$copy_num; i++));do + cat val_list.txt >> train_list.txt + done + cd ../../ + fi + + if [ ${#gpu_id} -le 1 ];then log_path="$SAVE_LOG/profiling_log" mkdir -p $log_path