Skip to content

GitLab

Commit 86ee4a57 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix mode

parent 2c9c7de1
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
View file @ 86ee4a57
...@@ -4,9 +4,9 @@ python:python3.7 ...@@ -4,9 +4,9 @@ python:python3.7
gpu_list:0|0,1 gpu_list:0|0,1
Global.use_gpu:True|True Global.use_gpu:True|True
Global.auto_cast:null Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=2 Global.epoch_num:benchmark_train=2
Global.save_model_dir:./output/ 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 Global.pretrained_model:null
train_model_name:latest train_model_name:latest
train_infer_img_dir:null train_infer_img_dir:null
......
test_tipc/prepare.sh
View file @ 86ee4a57
...@@ -21,6 +21,7 @@ model_name=$(func_parser_value "${lines[1]}") ...@@ -21,6 +21,7 @@ model_name=$(func_parser_value "${lines[1]}")
trainer_list=$(func_parser_value "${lines[14]}") trainer_list=$(func_parser_value "${lines[14]}")
if [ ${MODE} = "benchmark_train" ];then 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_benchmark" ]];then
rm -rf ./train_data/icdar2015 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 ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment