rename key
This commit is contained in:
parent
27b543ab13
commit
40f78f7531
|
@ -1,4 +1,4 @@
|
|||
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
|
||||
|
|
Loading…
Reference in New Issue