Merge pull request #23 from glenn-jocher/patch-2

Update ghost_net.py
iamhankai · Apr 11, 2020 · 8c601df · 8c601df
2 parents 2c90e67 + 9716675
commit 8c601df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ghost_net.py b/ghost_net.py
@@ -98,7 +98,7 @@ def __init__(self, inp, hidden_dim, oup, kernel_size, stride, use_se):
             self.shortcut = nn.Sequential()
         else:
             self.shortcut = nn.Sequential(
-                depthwise_conv(inp, inp, 3, stride, relu=True),
+                depthwise_conv(inp, inp, kernet_size, stride, relu=False),
                 nn.Conv2d(inp, oup, 1, 1, 0, bias=False),
                 nn.BatchNorm2d(oup),
             )