(self, layers, num_filters, **kwargs)
| 53 | |
| 54 | class audioEncoder(nn.Module): |
| 55 | def __init__(self, layers, num_filters, **kwargs): |
| 56 | super(audioEncoder, self).__init__() |
| 57 | block = SEBasicBlock |
| 58 | self.inplanes = num_filters[0] |
| 59 | |
| 60 | self.conv1 = nn.Conv2d(1, num_filters[0] , kernel_size=7, stride=(2, 1), padding=3, |
| 61 | bias=False) |
| 62 | self.bn1 = nn.BatchNorm2d(num_filters[0]) |
| 63 | self.relu = nn.ReLU(inplace=True) |
| 64 | |
| 65 | self.layer1 = self._make_layer(block, num_filters[0], layers[0]) |
| 66 | self.layer2 = self._make_layer(block, num_filters[1], layers[1], stride=(2, 2)) |
| 67 | self.layer3 = self._make_layer(block, num_filters[2], layers[2], stride=(2, 2)) |
| 68 | self.layer4 = self._make_layer(block, num_filters[3], layers[3], stride=(1, 1)) |
| 69 | out_dim = num_filters[3] * block.expansion |
| 70 | |
| 71 | for m in self.modules(): |
| 72 | if isinstance(m, nn.Conv2d): |
| 73 | nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu') |
| 74 | elif isinstance(m, nn.BatchNorm2d): |
| 75 | nn.init.constant_(m.weight, 1) |
| 76 | nn.init.constant_(m.bias, 0) |
| 77 | |
| 78 | def _make_layer(self, block, planes, blocks, stride=1): |
| 79 | downsample = None |
nothing calls this directly
no test coverage detected