updates

2020-02-04 21:22:20 -08:00 · 2020-02-04 21:22:20 -08:00 · e185719bd7
parent 888cad1e31
commit e185719bd7
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -268,7 +268,7 @@ class Darknet(nn.Module):
            x = [torch.cat(x, 0) for x in zip(*output)]
            return x[0], torch.cat(x[1:3], 1)  # scores, boxes: 3780x80, 3780x4
        else:
-            io, p = list(zip(*output))  # inference output, training output
+            io, p = zip(*output)  # inference output, training output
            return torch.cat(io, 1), p

    def fuse(self):