diff --git a/test.py b/test.py index 5d14c4e7..7472fa98 100644 --- a/test.py +++ b/test.py @@ -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