From ae96c979eb651825dfc35d5502c5721ca86a129d Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 28 Aug 2023 06:44:50 +0000
Subject: [PATCH] always log 'topk=1' when k < output_dims to ensure consistent
 log formatting

---
 ppcls/metric/metrics.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/metric/metrics.py b/ppcls/metric/metrics.py
index ec5be8ea..20282e22 100644
--- a/ppcls/metric/metrics.py
+++ b/ppcls/metric/metrics.py
@@ -54,8 +54,9 @@ class TopkAcc(AvgMetrics):
         metric_dict = dict()
         for idx, k in enumerate(self.topk):
             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)
+                metric_dict[f"top{k}"] = 1
                 self.avg_meters.pop(f"top{k}")
                 continue
             metric_dict[f"top{k}"] = paddle.metric.accuracy(x, label, k=k)
-- 
GitLab