Fix MixConv2d() remove shortcut + apply depthwise (#5410)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-10-30 13:38:51 +02:00
1 parent 7f9bbf0268
commit 5d4258fac5
3 files changed
+13 -12

No files matched your search

+1 -1
View File
@@ -113,7 +113,7 @@ class BottleneckCSP(nn.Module):
self.cv3 = nn.Conv2d(c_, c_, 1, 1, bias=False)
self.cv4 = Conv(2 * c_, c2, 1, 1)
self.bn = nn.BatchNorm2d(2 * c_) # applied to cat(cv2, cv3)
self.act = nn.LeakyReLU(0.1, inplace=True)
self.act = nn.SiLU()
self.m = nn.Sequential(*(Bottleneck(c_, c_, shortcut, g, e=1.0) for _ in range(n)))
def forward(self, x):