From d498193456a6676acbc383525b9dc30a23dddbbc Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 23 Jan 2020 15:15:53 -0800 Subject: [PATCH] updates --- models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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)