mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
Update metrics.py
This commit is contained in:
parent
9e83fb4c2f
commit
d6fe8c0c1b
@ -100,7 +100,6 @@ class RetriMetric(nn.Layer):
|
||||
assert isinstance(topk, (int, list, tuple))
|
||||
if isinstance(topk, int):
|
||||
topk = [topk]
|
||||
self.topk = topk
|
||||
for k in topk:
|
||||
metric_dict["recall{}".format(k)] = all_cmc[k - 1]
|
||||
if "mAP" in self.config.keys():
|
||||
|
Loading…
x
Reference in New Issue
Block a user