(self, input_dim: int = 16, output_dim: int = 8, stride: int = 1)
| 93 | |
| 94 | class DecoderBlock(nn.Module): |
| 95 | def __init__(self, input_dim: int = 16, output_dim: int = 8, stride: int = 1): |
| 96 | super().__init__() |
| 97 | self.block = nn.Sequential( |
| 98 | Snake1d(input_dim), |
| 99 | WNConvTranspose1d( |
| 100 | input_dim, |
| 101 | output_dim, |
| 102 | kernel_size=2 * stride, |
| 103 | stride=stride, |
| 104 | padding=math.ceil(stride / 2), |
| 105 | ), |
| 106 | ResidualUnit(output_dim, dilation=1), |
| 107 | ResidualUnit(output_dim, dilation=3), |
| 108 | ResidualUnit(output_dim, dilation=9), |
| 109 | ) |
| 110 | |
| 111 | def forward(self, x): |
| 112 | return self.block(x) |
nothing calls this directly
no test coverage detected