Skip to content

Commit

Permalink
fix distill
Browse files Browse the repository at this point in the history
  • Loading branch information
LDOUBLEV committed Apr 29, 2022
1 parent 7e5d605 commit 68da674
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_dml.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion doc/doc_ch/knowledge_distillation.md
Original file line number Diff line number Diff line change
Expand Up @@ -448,7 +448,7 @@ Architecture:
<a name="222"></a>
#### 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
Expand Down
2 changes: 1 addition & 1 deletion doc/doc_en/knowledge_distillation_en.md
Original file line number Diff line number Diff line change
Expand Up @@ -464,7 +464,7 @@ The key contains `backbone_out`, `neck_out`, `head_out`, and `value` is the tens

<a name="222"></a>
#### 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
Expand Down

0 comments on commit 68da674

Please sign in to comment.