diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml index bb6a196864b6e9e7525f2b5217f0c90ea2ca05a4..b75ca6fdc7e7543460ba2e561af27e4d4fbf3391 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml @@ -21,6 +21,7 @@ Architecture: model_type: det Models: Teacher: + pretrained: ./pretrain_models/ch_ppocr_server_v2.0_det_train/best_accuracy freeze_params: true return_all_feats: false model_type: det @@ -36,6 +37,7 @@ Architecture: name: DBHead k: 50 Student: + pretrained: freeze_params: false return_all_feats: false model_type: det @@ -52,6 +54,7 @@ Architecture: name: DBHead k: 50 Student2: + pretrained: freeze_params: false return_all_feats: false model_type: det diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml index d8d5135dd73ee438f76f5796b63e0dae4331402b..b4644244f849abb05d5501c7d2ed2bf92efe916e 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_distill.yml @@ -18,6 +18,7 @@ Global: Architecture: name: DistillationModel algorithm: Distillation + model_type: det Models: Student: pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml index bfbc3b6268cf521acb035be33ced9141046fc430..0553a65e448db21f196389f6abba7bf8808f0494 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml @@ -18,6 +18,7 @@ Global: Architecture: name: DistillationModel algorithm: Distillation + model_type: det Models: Student: pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained