Merge pull request #2151 from RainFrost1/develop

修复benchmark中bs过大时导致log过少的问题

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