Method
__init__
(self,
in_channels: int,
out_channels: int,
norm_cfg: ConfigType,
act_cfg: ConfigType = dict(type='Sigmoid'),
indice_key: Optional[str] = None)
Source from the content-addressed store, hash-verified
| 330 | """ |
| 331 | |
| 332 | def __init__(self, |
| 333 | in_channels: int, |
| 334 | out_channels: int, |
| 335 | norm_cfg: ConfigType, |
| 336 | act_cfg: ConfigType = dict(type='Sigmoid'), |
| 337 | indice_key: Optional[str] = None): |
| 338 | super().__init__() |
| 339 | |
| 340 | self.conv1 = SubMConv3d( |
| 341 | in_channels, |
| 342 | out_channels, |
| 343 | kernel_size=(3, 1, 1), |
| 344 | padding=1, |
| 345 | bias=False, |
| 346 | indice_key=indice_key) |
| 347 | self.bn1 = build_norm_layer(norm_cfg, out_channels)[1] |
| 348 | self.act1 = build_activation_layer(act_cfg) |
| 349 | |
| 350 | self.conv2 = SubMConv3d( |
| 351 | in_channels, |
| 352 | out_channels, |
| 353 | kernel_size=(1, 3, 1), |
| 354 | padding=1, |
| 355 | bias=False, |
| 356 | indice_key=indice_key) |
| 357 | self.bn2 = build_norm_layer(norm_cfg, out_channels)[1] |
| 358 | self.act2 = build_activation_layer(act_cfg) |
| 359 | |
| 360 | self.conv3 = SubMConv3d( |
| 361 | in_channels, |
| 362 | out_channels, |
| 363 | kernel_size=(1, 1, 3), |
| 364 | padding=1, |
| 365 | bias=False, |
| 366 | indice_key=indice_key) |
| 367 | self.bn3 = build_norm_layer(norm_cfg, out_channels)[1] |
| 368 | self.act3 = build_activation_layer(act_cfg) |
| 369 | |
| 370 | def forward(self, x: SparseConvTensor) -> SparseConvTensor: |
| 371 | """Forward pass.""" |
Tested by
no test coverage detected