fix mode
parent
2c9c7de1e3
commit
86ee4a5782
|
@ -4,9 +4,9 @@ python:python3.7
|
|||
gpu_list:0|0,1
|
||||
Global.use_gpu:True|True
|
||||
Global.auto_cast:null
|
||||
Global.epoch_num:lite_train_lite_infer=2
|
||||
Global.epoch_num:benchmark_train=2
|
||||
Global.save_model_dir:./output/
|
||||
Train.loader.batch_size_per_card:lite_train_lite_infer=16
|
||||
Train.loader.batch_size_per_card:benchmark_train=16
|
||||
Global.pretrained_model:null
|
||||
train_model_name:latest
|
||||
train_infer_img_dir:null
|
||||
|
|
|
@ -21,6 +21,7 @@ model_name=$(func_parser_value "${lines[1]}")
|
|||
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
|
||||
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
|
||||
|
|
Loading…
Reference in New Issue