From 89d0aa7164e3512c7f7576f6eda9295d1126298f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 12 Aug 2019 00:43:04 +0200 Subject: [PATCH] updates --- models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models.py b/models.py index 4c00d148..109b336c 100755 --- a/models.py +++ b/models.py @@ -201,8 +201,7 @@ class Darknet(nn.Module): x = torch.cat([layer_outputs[i] for i in layer_i], 1) # print(''), [print(layer_outputs[i].shape) for i in layer_i], print(x.shape) elif mtype == 'shortcut': - layer_i = int(mdef['from']) - x = layer_outputs[-1] + layer_outputs[layer_i] + x += layer_outputs[int(mdef['from'])] elif mtype == 'yolo': x = module(x, img_size) output.append(x)