(
self,
in_channels,
out_channels,
kernel_size=7,
dilation=1,
bias=False,
)
| 171 | |
| 172 | class NonCausalResUnit(nn.Module): |
| 173 | def __init__( |
| 174 | self, |
| 175 | in_channels, |
| 176 | out_channels, |
| 177 | kernel_size=7, |
| 178 | dilation=1, |
| 179 | bias=False, |
| 180 | ): |
| 181 | super().__init__() |
| 182 | self.activation = nn.ELU() |
| 183 | self.conv1 = NonCausalConv1d( |
| 184 | in_channels=in_channels, |
| 185 | out_channels=out_channels, |
| 186 | kernel_size=kernel_size, |
| 187 | stride=1, |
| 188 | dilation=dilation, |
| 189 | bias=bias, |
| 190 | ) |
| 191 | self.conv2 = Conv1d1x1(out_channels, out_channels, bias) |
| 192 | |
| 193 | def forward(self, x): |
| 194 | y = self.conv1(self.activation(x)) |
nothing calls this directly
no test coverage detected