rename and set MultiP for N1C4

pull/5277/head
LDOUBLEV 2022-01-27 10:47:50 +00:00
parent a99d4fbfe3
commit d8da4cb32e
2 changed files with 5 additions and 1 deletions

View File

@ -18,6 +18,10 @@ bash test_tipc/prepare.sh test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
```shell
# 运行格式bash test_tipc/benchmark_train.sh train_benchmark.txt mode params
bash test_tipc/benchmark_train.sh test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt benchmark_train dynamic_bs8_null_SingleP_DP_N1C1
# 单机多卡训练MultiP 表示多进程单卡训练用SingleP
# 运行格式bash test_tipc/benchmark_train.sh train_benchmark.txt mode params
bash test_tipc/benchmark_train.sh test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt benchmark_train dynamic_bs8_null_MultiP_DP_N1C4
```
params为test_tipc/benchmark_train.sh传入的参数包含模型类型、batchsize、fp精度、进程类型、运行模式以及分布式等信息。

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" ]];then
if [[ ${model_name} =~ "det_mv3_db_v2" ]];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