提交 40f78f75 编写于 作者: L LDOUBLEV

rename key

上级 27b543ab
train_model_list: det;benchmark/benchmark_det.yml
train_model_list: ocr_det
gpu_list: -1|0|0,1
auto_cast_list: False|True
trainer_list: norm|quant|prune
......
......@@ -55,10 +55,10 @@ checkpoints=$(func_parser "${lines[14]}")
for train_model in ${train_model_list[*]}; do
if [ ${train_model} = "det" ];then
if [ ${train_model} = "ocr_det" ];then
model_name="det"
yml_file="configs/det/det_mv3_db.yml"
elif [ ${train_model} = "rec" ];then
elif [ ${train_model} = "ocr_rec" ];then
model_name="rec"
yml_file="configs/rec/rec_mv3_none_bilstm_ctc.yml"
else
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册