diff --git a/benchmark/run_det.sh b/benchmark/run_det.sh index d035a6750b3d58934dec5416d581a1cbd95bc7e6..4cf32f77f0423c34f2a872510fe1a48c5e3743f7 100644 --- a/benchmark/run_det.sh +++ b/benchmark/run_det.sh @@ -17,14 +17,13 @@ bs_list=(8 16) for model_mode in ${model_mode_list[@]}; do for fp_item in ${fp_item_list[@]}; do 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} 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} 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