From 19616781772bc45f3d085201ca1585ddf9cd9ddc Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 27 Jan 2020 17:03:27 -0500 Subject: [PATCH] updates --- models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models.py b/models.py index ee7f7c47..cf5435b5 100755 --- a/models.py +++ b/models.py @@ -233,6 +233,8 @@ class Darknet(nn.Module): x = module(x) elif mtype == 'route': layers = [int(x) for x in mdef['layers'].split(',')] + if verbose: + print('route concatenating %s' % ([layer_outputs[i].shape for i in layers])) if len(layers) == 1: x = layer_outputs[layers[0]] else: