From 3bfbab7afd5850b4f21b73dd3184374f47eb1d98 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 9 Dec 2019 13:25:34 -0800
Subject: [PATCH] updates

---
 models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models.py b/models.py
index 066fe0ca..bff616e0 100755
--- 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))