rename mode

This commit is contained in:
LDOUBLEV 2022-01-24 09:32:55 +00:00
parent 94788c6d95
commit a530b69886
2 changed files with 4 additions and 4 deletions

View File

@ -1,8 +1,8 @@
===========================train_params===========================
model_name:det_mv3_db_v2.0_benchmark
model_name:det_mv3_db_v2.0
python:python
gpu_list:0|0,1
Global.use_gpu:True|True
gpu_list:0
Global.use_gpu:True
Global.auto_cast:null
Global.epoch_num:benchmark_train=2
Global.save_model_dir:./output/

View File

@ -22,7 +22,7 @@ trainer_list=$(func_parser_value "${lines[14]}")
if [ ${MODE} = "benchmark_train" ];then
pip install -r requirements.txt
if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then
if [[ ${model_name} =~ "det_mv3_db_v2.0" ]];then
rm -rf ./train_data/icdar2015
wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate