From 50af18d44cf87055b5732db55afed63d51c243b2 Mon Sep 17 00:00:00 2001 From: Topdu <784990967@qq.com> Date: Thu, 21 Jul 2022 02:53:51 +0000 Subject: [PATCH 1/2] fix vitstr yml bug --- configs/rec/rec_vitstr_none_ce.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/rec/rec_vitstr_none_ce.yml b/configs/rec/rec_vitstr_none_ce.yml index 13a7d7d1bc..ebe304fa4b 100644 --- a/configs/rec/rec_vitstr_none_ce.yml +++ b/configs/rec/rec_vitstr_none_ce.yml @@ -81,7 +81,7 @@ Train: Eval: dataset: name: LMDBDataSet - data_dir: ./train_data/data_lmdb_release/evaluaiton/ + data_dir: ./train_data/data_lmdb_release/evaluation/ transforms: - DecodeImage: # load image img_mode: BGR From ce600ab56f9e72f58b3fcb578d315b9db20ff402 Mon Sep 17 00:00:00 2001 From: Topdu <784990967@qq.com> Date: Thu, 21 Jul 2022 04:50:31 +0000 Subject: [PATCH 2/2] add abinet pretrained model --- configs/rec/rec_r45_abinet.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/rec/rec_r45_abinet.yml b/configs/rec/rec_r45_abinet.yml index 3c5e51229f..bc9048a200 100644 --- a/configs/rec/rec_r45_abinet.yml +++ b/configs/rec/rec_r45_abinet.yml @@ -8,7 +8,7 @@ Global: # evaluation is run every 2000 iterations eval_batch_step: [0, 2000] cal_metric_during_train: True - pretrained_model: + pretrained_model: ./pretrain_models/abinet_vl_pretrained checkpoints: save_inference_dir: use_visualdl: False