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

always log 'topk=1' when k < output_dims to ensure consistent log formatting

Browse Source
This commit is contained in:
gaotingquan 2023-08-28 06:44:50 +00:00 committed by Tingquan Gao
parent eddba911b1
commit ae96c979eb
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ppcls/metric/metrics.py
View File

@ -54,8 +54,9 @@ class TopkAcc(AvgMetrics):
metric_dict = dict() metric_dict = dict()
for idx, k in enumerate(self.topk): for idx, k in enumerate(self.topk):
if output_dims < k: if output_dims < k:
msg = f"The output dims({output_dims}) is less than k({k}), and the argument {k} of Topk has been removed." msg = f"The output dims({output_dims}) is less than k({k}), so the Top-{k} metric is meaningless."
logger.warning(msg) logger.warning(msg)
metric_dict[f"top{k}"] = 1
self.avg_meters.pop(f"top{k}") self.avg_meters.pop(f"top{k}")
continue continue
metric_dict[f"top{k}"] = paddle.metric.accuracy(x, label, k=k) metric_dict[f"top{k}"] = paddle.metric.accuracy(x, label, k=k)