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
|
```shell
|
||||||
# 运行格式:bash test_tipc/benchmark_train.sh train_benchmark.txt mode params
|
# 运行格式: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
|
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精度、进程类型、运行模式以及分布式等信息。
|
params为test_tipc/benchmark_train.sh传入的参数,包含:模型类型、batchsize、fp精度、进程类型、运行模式以及分布式等信息。
|
||||||
|
|
|
@ -22,7 +22,7 @@ trainer_list=$(func_parser_value "${lines[14]}")
|
||||||
|
|
||||||
if [ ${MODE} = "benchmark_train" ];then
|
if [ ${MODE} = "benchmark_train" ];then
|
||||||
pip install -r requirements.txt
|
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
|
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
|
||||||
wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --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