This commit is contained in:
Glenn Jocher 2019-12-16 22:18:26 -08:00
parent 9c03ac3b74
commit ecce92d5d8
1 changed files with 1 additions and 1 deletions

View File

@ -97,7 +97,7 @@ def test(cfg,
if pred is None:
if nl:
stats.append(([], torch.Tensor(), torch.Tensor(), tcls))
stats.append((torch.zeros(0, 1), torch.Tensor(), torch.Tensor(), tcls))
continue
# Append to text file