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

Empty val batch CUDA device fix (#7539)

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