cml config

b6cd6047 · LDOUBLEV · 293e85bc · b6cd6047
Commit b6cd6047 authored Apr 27, 2022 by LDOUBLEV
Show whitespace changes
Inline Side-by-side

Showing with 53 additions and 50 deletions

configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +53 -50

No files found.
--- 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,9 +19,10 @@ Global:
  distributed: true

 Architecture:
+  name: DistillationModel
+  algorithm: Distillation
  model_type: det
-  algorithm: DB
-NetWorks:
+  Models:
    Student:
      model_type: det
      algorithm: DB
@@ -56,6 +57,8 @@ NetWorks:
        k: 50
    Teacher:
      pretrained: ./inference/ch_ppocr_resnet50/student
+      freeze_params: true
+      return_all_feats: false
      model_type: det
      algorithm: DB
      Backbone: