未验证 提交 d6fe8c0c 编写于 作者: B Bin Lu 提交者: GitHub

Update metrics.py

上级 9e83fb4c
......@@ -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():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册