This commit is contained in:
Glenn Jocher
2019-12-16 22:18:26 -08:00
parent 9c03ac3b74
commit ecce92d5d8
+1 -1
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