uyzhang commited on
Commit
f532bc6
1 Parent(s): 9291daa

GhostConv() bug fix (#1176)

Browse files
Files changed (1) hide show
  1. 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)