From 8457594664db1f18919cf6674bafdb3b0214a585 Mon Sep 17 00:00:00 2001 From: Yi_Zhang Date: Tue, 20 Oct 2020 19:42:11 +0800 Subject: [PATCH] GhostConv() bug fix (#1176) --- models/experimental.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/experimental.py b/models/experimental.py index 1b99ce47db53..0b61027b9d2f 100644 --- 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)