diff --git a/benchmark/run_det.sh b/benchmark/run_det.sh index ebecf88c0ca8daee6ff890d6238b250b88991b69..be0c141f7ee168d10eebb6efb57158d18ed02f72 100644 --- a/benchmark/run_det.sh +++ b/benchmark/run_det.sh @@ -21,13 +21,14 @@ for model_mode in ${model_mode_list[@]}; do bs_list=(8 16) fi for bs_item in ${bs_list[@]}; do - log_name=ocr_${model_mode}_bs${bs_item}_${fp_item} echo "index is speed, 1gpus, begin, ${model_name}" run_mode=sp + log_name=ocr_${model_mode}_bs${bs_item}_${fp_item}_${run_mode} CUDA_VISIBLE_DEVICES=0 bash benchmark/run_benchmark_det.sh ${run_mode} ${bs_item} ${fp_item} 1 ${model_mode} | tee ${log_path}/${log_name}_speed_1gpus 2>&1 # (5min) sleep 60 echo "index is speed, 8gpus, run_mode is multi_process, begin, ${model_name}" run_mode=mp + log_name=ocr_${model_mode}_bs${bs_item}_${fp_item}_${run_mode} CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash benchmark/run_benchmark_det.sh ${run_mode} ${bs_item} ${fp_item} 2 ${model_mode} | tee ${log_path}/${log_name}_speed_8gpus8p 2>&1 sleep 60 done