fix: fix key error in distillation

ppcls/engine/evaluation/classification.py

@@ -78,7 +78,13 @@ def classification_eval(engine, epoch_id=0):
                 labels = paddle.concat(label_list, 0)
 
                 if isinstance(out, dict):
-                    out = out["logits"]
+                    if "logits" in out:
+                        out = out["logits"]
+                    elif "Student" in out:
+                        out = out["Student"]
+                    else:
+                        msg = "Error: Wrong key in out!"
+                        raise Exception(msg)
                 if isinstance(out, list):
                     pred = []
                     for x in out:

ppcls/metric/metrics.py

@@ -222,7 +222,8 @@ class DistillationTopkAcc(TopkAcc):
         self.feature_key = feature_key
 
     def forward(self, x, label):
-        x = x[self.model_key]
+        if isinstance(x, dict):
+            x = x[self.model_key]
         if self.feature_key is not None:
             x = x[self.feature_key]
         return super().forward(x, label)