From 207cf14df46776f753059888178325d03b8d9f41 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 9 Mar 2020 18:03:34 -0700 Subject: [PATCH] updates --- models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/models.py b/models.py index c71aaf13..a326d7d6 100755 --- a/models.py +++ b/models.py @@ -41,6 +41,9 @@ def create_modules(module_defs, img_size, arc): elif mdef['activation'] == 'swish': modules.add_module('activation', Swish()) + if not bn: # detection output layer + routs.append(i) + elif mdef['type'] == 'maxpool': size = mdef['size'] stride = mdef['stride']