mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

add fleet tests

Browse Source
This commit is contained in:
stephon 2022-02-09 09:24:49 +00:00
parent 0979bb76c2
commit 8a44a90396
6 changed files with 41 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test_tipc/supplementary/test_tipc/test_train_python.sh
View File

@ -35,7 +35,6 @@ use_share_conv_key=$(func_parser_key "${lines[13]}")
use_share_conv_list=$(func_parser_value "${lines[13]}") use_share_conv_list=$(func_parser_value "${lines[13]}")
run_train_py=$(func_parser_value "${lines[14]}") run_train_py=$(func_parser_value "${lines[14]}")
LOG_PATH="./test_tipc/extra_output" LOG_PATH="./test_tipc/extra_output"
mkdir -p ${LOG_PATH} mkdir -p ${LOG_PATH}
status_log="${LOG_PATH}/results_python.log" status_log="${LOG_PATH}/results_python.log"
@ -98,6 +97,8 @@ if [ ${MODE} = "lite_train_lite_infer" ] || [ ${MODE} = "whole_train_whole_infer
cmd="${python} ${run_train_py} ${set_use_gpu} ${set_save_model} ${set_epoch} ${set_pretrain} ${set_checkpoints} ${set_autocast} ${set_batchsize} ${set_use_custom_op} ${set_model_type} ${set_use_share_conv} ${set_amp_config}" cmd="${python} ${run_train_py} ${set_use_gpu} ${set_save_model} ${set_epoch} ${set_pretrain} ${set_checkpoints} ${set_autocast} ${set_batchsize} ${set_use_custom_op} ${set_model_type} ${set_use_share_conv} ${set_amp_config}"
elif [ ${#ips} -le 26 ];then # train with multi-gpu elif [ ${#ips} -le 26 ];then # train with multi-gpu
cmd="${python} -m paddle.distributed.launch --gpus=${gpu} ${run_train_py} ${set_use_gpu} ${set_save_model} ${set_epoch} ${set_pretrain} ${set_checkpoints} ${set_autocast} ${set_batchsize} ${set_use_custom_op} ${set_model_type} ${set_use_share_conv} ${set_amp_config}" cmd="${python} -m paddle.distributed.launch --gpus=${gpu} ${run_train_py} ${set_use_gpu} ${set_save_model} ${set_epoch} ${set_pretrain} ${set_checkpoints} ${set_autocast} ${set_batchsize} ${set_use_custom_op} ${set_model_type} ${set_use_share_conv} ${set_amp_config}"
else
cmd="${python} -m paddle.distributed.launch --ips=${ips} --gpus=${gpu} ${run_train_py} ${set_use_gpu} ${set_save_model} ${set_epoch} ${set_pretrain} ${set_checkpoints} ${set_autocast} ${set_batchsize} ${set_use_custom_op} ${set_model_type} ${set_use_share_conv} ${set_amp_config}"
fi fi
# run train # run train

4
test_tipc/supplementary/test_tipc/train_infer_python_FPGM.txt
View File

@ -4,9 +4,9 @@ python:python3.7
gpu_list:0|0,1 gpu_list:0|0,1
use_gpu:True|True use_gpu:True|True
AMP.use_amp:True|False AMP.use_amp:True|False
epoch:lite_train_lite_infer=20|whole_train_whole_infer=1000 epoch:lite_train_lite_infer=2|whole_train_whole_infer=1000
save_model_dir:./output/ save_model_dir:./output/
TRAIN.batch_size:lite_train_lite_infer=2|whole_train_whole_infer=4 TRAIN.batch_size:lite_train_lite_infer=1280|whole_train_whole_infer=1280
pretrained_model:null pretrained_model:null
checkpoints:null checkpoints:null
use_custom_relu:False|True use_custom_relu:False|True

17
test_tipc/supplementary/test_tipc/train_infer_python_FPGM_fleet.txt Normal file
View File

@ -0,0 +1,17 @@
===========================train_params===========================
model_name:ch_PPOCRv2_det
python:python3.7
gpu_list:xx.xx.xx.xx,yy.yy.yy.yy;0,1
use_gpu:True
AMP.use_amp:True|False
epoch:lite_train_lite_infer=2|whole_train_whole_infer=1000
save_model_dir:./output/
TRAIN.batch_size:lite_train_lite_infer=1280|whole_train_whole_infer=1280
pretrained_model:null
checkpoints:null
use_custom_relu:False|True
model_type:cls|cls_distill|cls_distill_multiopt
MODEL.siamese:False|True
norm_train:train.py -c mv3_large_x0_5.yml -o prune_train=True
quant_train:False
prune_train:False

4
test_tipc/supplementary/test_tipc/train_infer_python_PACT.txt
View File

@ -4,9 +4,9 @@ python:python3.7
gpu_list:0|0,1 gpu_list:0|0,1
use_gpu:True|True use_gpu:True|True
AMP.use_amp:True|False AMP.use_amp:True|False
epoch:lite_train_lite_infer=20|whole_train_whole_infer=1000 epoch:lite_train_lite_infer=2|whole_train_whole_infer=1000
save_model_dir:./output/ save_model_dir:./output/
TRAIN.batch_size:lite_train_lite_infer=2|whole_train_whole_infer=4 TRAIN.batch_size:lite_train_lite_infer=1280|whole_train_whole_infer=1280
pretrained_model:null pretrained_model:null
checkpoints:null checkpoints:null
use_custom_relu:False|True use_custom_relu:False|True

17
test_tipc/supplementary/test_tipc/train_infer_python_PACT_fleet.txt Normal file
View File

@ -0,0 +1,17 @@
===========================train_params===========================
model_name:ch_PPOCRv2_det
python:python3.7
gpu_list:xx.xx.xx.xx,yy.yy.yy.yy;0,1
use_gpu:True
AMP.use_amp:True|False
epoch:lite_train_lite_infer=2|whole_train_whole_infer=1000
save_model_dir:./output/
TRAIN.batch_size:lite_train_lite_infer=1280|whole_train_whole_infer=1280
pretrained_model:null
checkpoints:null
use_custom_relu:False|True
model_type:cls|cls_distill|cls_distill_multiopt
MODEL.siamese:False|True
norm_train:train.py -c mv3_large_x0_5.yml -o quant_train=True
quant_train:False
prune_train:False

3
test_tipc/supplementary/test_tipc/train_infer_python_fleet.txt
View File

@ -12,7 +12,6 @@ checkpoints:null
use_custom_relu:False|True use_custom_relu:False|True
model_type:cls|cls_distill|cls_distill_multiopt model_type:cls|cls_distill|cls_distill_multiopt
MODEL.siamese:False|True MODEL.siamese:False|True
norm_train:train.py -c mv3_large_x0_5.yml -o norm_train: train.py -c mv3_large_x0_5.yml -o
quant_train:False quant_train:False
prune_train:False prune_train:False