diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml index 95da6d24c2ce947f7bc422e082338b1e26f0efce..078edc2fea539d9b345d242308760287a1b818b3 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml @@ -19,57 +19,60 @@ Global: distributed: true Architecture: + name: DistillationModel + algorithm: Distillation model_type: det - algorithm: DB -NetWorks: - Student: - model_type: det - algorithm: DB - Transform: null - Backbone: - name: MobileNetV3 - scale: 0.5 - model_name: large - disable_se: true - Neck: - name: CAFPN - out_channels: 96 - shortcut: True - Head: - name: DBHead - k: 50 - Student2: - model_type: det - algorithm: DB - Transform: null - Backbone: - name: MobileNetV3 - scale: 0.5 - model_name: large - disable_se: true - Neck: - name: CAFPN - out_channels: 96 - shortcut: True - Head: - name: DBHead - k: 50 - Teacher: - pretrained: ./inference/ch_ppocr_resnet50/student - model_type: det - algorithm: DB - Backbone: - name: ResNet - in_channels: 3 - layers: 50 - Neck: - name: FEPAN - out_channels: 256 - Head: - name: DBHead - kernel_list: [7,2,2] - k: 50 - + Models: + Student: + model_type: det + algorithm: DB + Transform: null + Backbone: + name: MobileNetV3 + scale: 0.5 + model_name: large + disable_se: true + Neck: + name: CAFPN + out_channels: 96 + shortcut: True + Head: + name: DBHead + k: 50 + Student2: + model_type: det + algorithm: DB + Transform: null + Backbone: + name: MobileNetV3 + scale: 0.5 + model_name: large + disable_se: true + Neck: + name: CAFPN + out_channels: 96 + shortcut: True + Head: + name: DBHead + k: 50 + Teacher: + pretrained: ./inference/ch_ppocr_resnet50/student + freeze_params: true + return_all_feats: false + model_type: det + algorithm: DB + Backbone: + name: ResNet + in_channels: 3 + layers: 50 + Neck: + name: FEPAN + out_channels: 256 + Head: + name: DBHead + kernel_list: [7,2,2] + k: 50 + Loss: name: CombinedLoss loss_config_list: