From 7da4cf719a2757c35040addedeaa6fff1c999aea Mon Sep 17 00:00:00 2001 From: JYChen Date: Fri, 15 Oct 2021 10:15:00 +0800 Subject: [PATCH] add pose/solov2 model training benchmark (#4303) --- benchmark/run_all.sh | 3 +++ benchmark/run_benchmark.sh | 5 ++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh index 50370afb0..eeacf5cb8 100644 --- a/benchmark/run_all.sh +++ b/benchmark/run_all.sh @@ -19,6 +19,9 @@ for model_name in ${model_name_list[@]}; do fcos) bs_list=(2 8) ;; deformable_detr) bs_list=(2) ;; gfl) bs_list=(2 8) ;; + hrnet) bs_list=(64 160) ;; + higherhrnet) bs_list=(20 24) ;; + solov2) bs_list=(2 4) ;; *) echo "wrong model_name"; exit 1; esac for bs_item in ${bs_list[@]} diff --git a/benchmark/run_benchmark.sh b/benchmark/run_benchmark.sh index 01e01157e..14414d049 100644 --- a/benchmark/run_benchmark.sh +++ b/benchmark/run_benchmark.sh @@ -26,12 +26,15 @@ function _train(){ fcos) model_yml="configs/fcos/fcos_r50_fpn_1x_coco.yml" ;; deformable_detr) model_yml="configs/deformable_detr/deformable_detr_r50_1x_coco.yml" ;; gfl) model_yml="configs/gfl/gfl_r50_fpn_1x_coco.yml" ;; + hrnet) model_yml="configs/keypoint/hrnet/hrnet_w32_256x192.yml" ;; + higherhrnet) model_yml="configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml" ;; + solov2) model_yml="configs/solov2/solov2_r50_fpn_1x_coco.yml" ;; *) echo "Undefined model_name"; exit 1; esac set_batch_size="TrainReader.batch_size=${batch_size}" set_max_epoch="epoch=${max_epoch}" - set_log_iter="log_iter=10" + set_log_iter="log_iter=1" if [ ${fp_item} = "fp16" ]; then set_fp_item="--fp16" else -- GitLab