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

Fix AP calculation bug #8464 (#8484)

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