From 68da674dfd91e3ce46d7ff6e9194e659680a4c1a Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 29 Apr 2022 10:23:51 +0800 Subject: [PATCH] fix distill --- configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml | 2 +- doc/doc_ch/knowledge_distillation.md | 2 +- doc/doc_en/knowledge_distillation_en.md | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml index fe92b8bd..3f30ada1 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml @@ -35,7 +35,7 @@ Architecture: name: DBHead kernel_list: [7,2,2] k: 50 - Teacher: + Student2: return_all_feats: false model_type: det algorithm: DB diff --git a/doc/doc_ch/knowledge_distillation.md b/doc/doc_ch/knowledge_distillation.md index da79e32b..7c79f885 100644 --- a/doc/doc_ch/knowledge_distillation.md +++ b/doc/doc_ch/knowledge_distillation.md @@ -448,7 +448,7 @@ Architecture: #### 2.2.2 损失函数 -检测ch_PP-OCRv3_det_cml.yml蒸馏损失函数配置如下所示。相比较于ch_PP-OCRv3_det_distill.yml的损失函数配置，cml蒸馏的损失函数配置做了3个改动： +检测ch_PP-OCRv3_det_cml.yml蒸馏损失函数配置如下所示。 ```yaml Loss: name: CombinedLoss diff --git a/doc/doc_en/knowledge_distillation_en.md b/doc/doc_en/knowledge_distillation_en.md index faf7213a..8ba7bf91 100755 --- a/doc/doc_en/knowledge_distillation_en.md +++ b/doc/doc_en/knowledge_distillation_en.md @@ -464,7 +464,7 @@ The key contains `backbone_out`, `neck_out`, `head_out`, and `value` is the tens #### 2.2.2 Loss Function -The distillation loss function configuration(`ch_PP-OCRv3_det_cml.yml`) is shown below. Compared with the loss function configuration of ch_PP-OCRv3_det_distill.yml, there are three changes: +The distillation loss function configuration(`ch_PP-OCRv3_det_cml.yml`) is shown below. ```yaml Loss: name: CombinedLoss -- GitLab