diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh index 0a6e3389b07d18e3ff46a620e4a7c70c41a611aa..125c61cb4103521e16898d2325a22978698a7fa8 100644 --- a/test_tipc/benchmark_train.sh +++ b/test_tipc/benchmark_train.sh @@ -274,7 +274,12 @@ for batch_size in ${batch_size_list[*]}; do speed_log_name="${repo_name}_${model_name}_bs${batch_size}_${precision}_${run_mode}_${device_num}_${to_static}speed" func_sed_params "$FILENAME" "${line_gpuid}" "$gpu_id" # sed used gpu_id func_sed_params "$FILENAME" "${line_profile}" "null" # sed --profile_option as null - cmd="timeout 5m bash test_tipc/test_train_inference_python.sh ${FILENAME} benchmark_train > ${log_path}/${log_name} 2>&1 " + if [[ ${device_num} = "N4C32" ]];then + duration=10m + else + duration=5m + fi + cmd="timeout ${duration} bash test_tipc/test_train_inference_python.sh ${FILENAME} benchmark_train > ${log_path}/${log_name} 2>&1 " echo $cmd job_bt=`date '+%Y%m%d%H%M%S'` eval ${cmd}