fix: dml -> udml

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_distillation.yaml

@@ -26,27 +26,42 @@ Arch:
   pretrained_list:
   # if not null, its lengths should be same as models
   freeze_params_list:
-  - True
+  - False
   - False
   use_sync_bn: True
   models:
     - Teacher:
-        name: ResNet101_vd
+        name: PPLCNet_x1_0
         class_num: *class_num
-        pretrained: pretrained/teacher_ResNet101_vd_0/ResNet101_vd/best_model
+        pretrained: True
+        use_ssld: True
+        return_stages: True
+        return_patterns: ["blocks3", "blocks4", "blocks5", "blocks6"]
     - Student:
         name: PPLCNet_x1_0
         class_num: *class_num
         pretrained: True
         use_ssld: True
+        return_stages: True
+        return_patterns: ["blocks3", "blocks4", "blocks5", "blocks6"]
 
   infer_model_name: "Student"
  
 # loss function config for traing/eval process
 Loss:
   Train:
+    - DistillationGTCELoss:
+        weight: 1.0
+        key: logits
+        model_names: ["Student", "Teacher"]
     - DistillationDMLLoss:
         weight: 1.0
+        key: logits
+        model_name_pairs:
+        - ["Student", "Teacher"]
+    - DistillationDistanceLoss:
+        weight: 1.0
+        key: "blocks4"
         model_name_pairs:
         - ["Student", "Teacher"]
   Eval:

ppcls/configs/PULC/safety_helmet/search.yaml

@@ -30,11 +30,7 @@ search_dict:
       - [0.0, 0.4, 0.4, 0.8, 0.8, 1.0]
       - [1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
 teacher:
-  rm_keys:
-    - Arch.lr_mult_list
-  search_values:
-    - ResNet101_vd
-    - ResNet50_vd
+  algorithm: "udml"
 final_replace:
   Arch.lr_mult_list: Arch.models.1.Student.lr_mult_list