From d8da4cb32e741a679e24085d4683cb1b8583e9f6 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 27 Jan 2022 10:47:50 +0000 Subject: [PATCH] rename and set MultiP for N1C4 --- test_tipc/docs/benchmark_train.md | 4 ++++ test_tipc/prepare.sh | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/test_tipc/docs/benchmark_train.md b/test_tipc/docs/benchmark_train.md index 0dd6b0e8..7970cdf6 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 ef0272e0..4be74ab7 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 -- GitLab