Method
__init__
(
self,
inplanes,
planes,
stride=1,
dilation=1,
downsample=None,
conv_type=ConvType.HYPERCUBE,
reduction=16,
D=-1,
)
Source from the content-addressed store, hash-verified
| 26 | |
| 27 | class SEBasicBlock(BasicBlock): |
| 28 | def __init__( |
| 29 | self, |
| 30 | inplanes, |
| 31 | planes, |
| 32 | stride=1, |
| 33 | dilation=1, |
| 34 | downsample=None, |
| 35 | conv_type=ConvType.HYPERCUBE, |
| 36 | reduction=16, |
| 37 | D=-1, |
| 38 | ): |
| 39 | super().__init__( |
| 40 | inplanes, |
| 41 | planes, |
| 42 | stride=stride, |
| 43 | dilation=dilation, |
| 44 | downsample=downsample, |
| 45 | conv_type=conv_type, |
| 46 | D=D, |
| 47 | ) |
| 48 | self.se = SELayer(planes, reduction=reduction, D=D) |
| 49 | |
| 50 | def forward(self, x): |
| 51 | residual = x |
Callers
nothing calls this directly
Tested by
no test coverage detected