updates

2020-03-09 18:03:34 -07:00 · 2020-03-09 18:03:34 -07:00 · 207cf14df4
parent 204594f299
commit 207cf14df4
1 changed files with 3 additions and 0 deletions
--- 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']