Unverified 提交 813eba85 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Empty val batch CUDA device fix (#7539)

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