From 6c68808a3ffd9c8221674a4ac043d476750d001b Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Sun, 1 May 2022 19:54:22 +0800
Subject: [PATCH] fix dml (#6067)

---
 configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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 0553a65e..f3ad9664 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
@@ -37,7 +37,7 @@ Architecture:
       Head:
         name: DBHead
         k: 50
-    Student2:
+    Teacher:
       pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained
       freeze_params: false
       return_all_feats: false
@@ -62,15 +62,15 @@ Loss:
   loss_config_list:
   - DistillationDMLLoss:
       model_name_pairs:
-      - ["Student", "Student2"]
+      - ["Student", "Teacher"]
       maps_name: "thrink_maps"
       weight: 1.0
       act: "softmax"
-      model_name_pairs: ["Student", "Student2"]
+      model_name_pairs: ["Student", "Teacher"]
       key: maps
   - DistillationDBLoss:
       weight: 1.0
-      model_name_list: ["Student", "Student2"]
+      model_name_list: ["Student", "Teacher"]
       # key: maps
       name: DBLoss
       balance_loss: true
@@ -94,7 +94,7 @@ Optimizer:
 
 PostProcess:
   name: DistillationDBPostProcess
-  model_name: ["Student", "Student2"]
+  model_name: ["Student", "Teacher"]
   key: head_out
   thresh: 0.3
   box_thresh: 0.6
-- 
GitLab