updates

2019-12-09 13:25:34 -08:00 · 2019-12-09 13:25:34 -08:00 · 3bfbab7afd
parent 07c1fafba8
commit 3bfbab7afd
1 changed files with 1 additions and 0 deletions
--- a/models.py
+++ b/models.py
@ -30,6 +30,7 @@ def create_modules(module_defs, img_size, arc):
                                                   kernel_size=size,
                                                   stride=stride,
                                                   padding=pad,
+                                                   groups=int(mdef['groups']) if 'groups' in mdef else 1,
                                                   bias=not bn))
            if bn:
                modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.1))