PyTorch Hub and autoShape update (#1415)

* PyTorch Hub and autoShape update

* comment x for imgs

* reduce comment
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-16 23:09:55 +01:00
1 parent 92c9b72832
commit f5429260ca
5 files changed
+84 -35

No files matched your search

+1 -1
View File
@@ -126,7 +126,7 @@ def test(data,
tcls = labels[:, 0].tolist() if nl else [] # target class
seen += 1
if pred is None:
if len(pred) == 0:
if nl:
stats.append((torch.zeros(0, niou, dtype=torch.bool), torch.Tensor(), torch.Tensor(), tcls))
continue