提交 8a44a903 编写于 作者: S stephon

add fleet tests

上级 0979bb76
...@@ -35,7 +35,6 @@ use_share_conv_key=$(func_parser_key "${lines[13]}") ...@@ -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 ...@@ -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,9 +4,9 @@ python:python3.7 ...@@ -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
......
===========================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,9 +4,9 @@ python:python3.7 ...@@ -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
......
===========================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
...@@ -12,7 +12,6 @@ checkpoints:null ...@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册