Add DWConvClass() (#4274)

* Add `DWConvClass()`

* Cleanup

* Cleanup2
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-08-02 15:36:30 +02:00
1 parent 9c6732f61c
commit 587c4b4b81
3 files changed
+12 -5

No files matched your search

+1 -1
View File
@@ -72,7 +72,7 @@ class GhostBottleneck(nn.Module):
class MixConv2d(nn.Module):
# Mixed Depthwise Conv https://arxiv.org/abs/1907.09595
# Mixed Depth-wise Conv https://arxiv.org/abs/1907.09595
def __init__(self, c1, c2, k=(1, 3), s=1, equal_ch=True):
super().__init__()
groups = len(k)