Update metrics.py

pull/817/head
Bin Lu 2021-06-10 10:41:58 +08:00 committed by GitHub
parent 9e83fb4c2f
commit d6fe8c0c1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions

View File

@ -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():