Unverified 提交 39d7a936 authored 作者: UnglvKitDe's avatar UnglvKitDe 提交者: GitHub

Fix AP calculation bug #8464 (#8484)

上级 dd28df98
......@@ -227,7 +227,7 @@ def run(
if npr == 0:
if nl:
stats.append((correct, *torch.zeros((3, 0), device=device)))
stats.append((correct, *torch.zeros((2, 0), device=device), labels[:, 0]))
continue
# Predictions
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论