fix_log_name

pull/4797/head
hysunflower 2021-11-30 07:15:11 +00:00
parent 0a52b052a0
commit 4f1b432a1b
1 changed files with 5 additions and 1 deletions

View File

@ -13,9 +13,13 @@ wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dyg
model_mode_list=(det_res18_db_v2.0 det_r50_vd_east det_r50_vd_pse)
fp_item_list=(fp32)
bs_list=(8 16)
for model_mode in ${model_mode_list[@]}; do
for fp_item in ${fp_item_list[@]}; do
if [ ${model_mode} == "det_res18_db_v2.0" ]; then
bs_list=(8 16)
else
bs_list=(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}"