updates

2020-01-23 15:15:53 -08:00 · 2020-01-23 15:15:53 -08:00 · d498193456
parent 07c8a03aa0
commit d498193456
1 changed files with 1 additions and 2 deletions
--- a/models.py
+++ b/models.py
@ -222,8 +222,7 @@ class Darknet(nn.Module):

    def forward(self, x, var=None):
        img_size = x.shape[-2:]
-        layer_outputs = []
-        output = []
+        output, layer_outputs = [], []
        verbose = False
        if verbose:
            print('0', x.shape)