Yi_Zhang
commited on
GhostConv() bug fix (#1176)
Browse files- models/experimental.py +2 -2
models/experimental.py
CHANGED
@@ -67,8 +67,8 @@ class GhostConv(nn.Module):
|
|
67 |
def __init__(self, c1, c2, k=1, s=1, g=1, act=True): # ch_in, ch_out, kernel, stride, groups
|
68 |
super(GhostConv, self).__init__()
|
69 |
c_ = c2 // 2 # hidden channels
|
70 |
-
self.cv1 = Conv(c1, c_, k, s, g, act)
|
71 |
-
self.cv2 = Conv(c_, c_, 5, 1, c_, act)
|
72 |
|
73 |
def forward(self, x):
|
74 |
y = self.cv1(x)
|
|
|
67 |
def __init__(self, c1, c2, k=1, s=1, g=1, act=True): # ch_in, ch_out, kernel, stride, groups
|
68 |
super(GhostConv, self).__init__()
|
69 |
c_ = c2 // 2 # hidden channels
|
70 |
+
self.cv1 = Conv(c1, c_, k, s, None, g, act)
|
71 |
+
self.cv2 = Conv(c_, c_, 5, 1, None, c_, act)
|
72 |
|
73 |
def forward(self, x):
|
74 |
y = self.cv1(x)
|