GhostConv() bug fix (#1176)

ultralytics · Oct 20, 2020 · f532bc6 · f532bc6
1 parent 9291daa
commit f532bc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/experimental.py b/models/experimental.py
@@ -67,8 +67,8 @@ class GhostConv(nn.Module):
     def __init__(self, c1, c2, k=1, s=1, g=1, act=True):  # ch_in, ch_out, kernel, stride, groups
         super(GhostConv, self).__init__()
         c_ = c2 // 2  # hidden channels
-        self.cv1 = Conv(c1, c_, k, s, g, act)
-        self.cv2 = Conv(c_, c_, 5, 1, c_, act)
+        self.cv1 = Conv(c1, c_, k, s, None, g, act)
+        self.cv2 = Conv(c_, c_, 5, 1, None, c_, act)
 
     def forward(self, x):
         y = self.cv1(x)