Fix MixConv2d() remove shortcut + apply depthwise (#5410)
This commit is contained in:
1 parent
7f9bbf0268
commit
5d4258fac5
3 files changed
+13
-12
No files matched your search
+1
-1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user