mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix: fix key error in distillation

Browse Source
This commit is contained in:
gaotingquan 2021-12-15 07:53:09 +00:00 committed by Tingquan Gao
parent 26f329c20a
commit a9a2e2d372
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppcls/engine/evaluation/classification.py
View File

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

1
ppcls/metric/metrics.py
View File

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