diff --git a/models.py b/models.py index 4be680ee..ee7f7c47 100755 --- 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)