diff --git a/test_tipc/docs/benchmark_train.md b/test_tipc/docs/benchmark_train.md index 0dd6b0e823151150ed250a6f63ce11aa2640531f..7970cdf6e9308261cd45f81b0f75c4951217fcec 100644 --- a/test_tipc/docs/benchmark_train.md +++ b/test_tipc/docs/benchmark_train.md @@ -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精度、进程类型、运行模式以及分布式等信息。 diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index ef0272e0d5a2960e11e5d95c968a141ab0ba65cd..4be74ab700627b9d877b01d1c1360c0d256b3200 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -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