Method
__init__
(self, in_channels, out_channels, ksize, stride=1, act="silu")
Source from the content-addressed store, hash-verified
| 58 | """Depthwise Conv + Conv""" |
| 59 | |
| 60 | def __init__(self, in_channels, out_channels, ksize, stride=1, act="silu"): |
| 61 | super().__init__() |
| 62 | self.dconv = BaseConv( |
| 63 | in_channels, |
| 64 | in_channels, |
| 65 | ksize=ksize, |
| 66 | stride=stride, |
| 67 | groups=in_channels, |
| 68 | act=act, |
| 69 | ) |
| 70 | self.pconv = BaseConv( |
| 71 | in_channels, out_channels, ksize=1, stride=1, groups=1, act=act |
| 72 | ) |
| 73 | |
| 74 | def forward(self, x): |
| 75 | x = self.dconv(x) |
Callers
nothing calls this directly
Tested by
no test coverage detected