rename and set MultiP for N1C4
parent
a99d4fbfe3
commit
d8da4cb32e
|
@ -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精度、进程类型、运行模式以及分布式等信息。
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue