(self, channels, kernel_size=3, drop_rate=0.0)
| 32 | |
| 33 | class ResidualBlock(nn.Module): |
| 34 | def __init__(self, channels, kernel_size=3, drop_rate=0.0): |
| 35 | super().__init__() |
| 36 | self.conv1 = ConvBlock(channels, channels, kernel_size, padding=kernel_size // 2, drop_rate=drop_rate) |
| 37 | self.conv2 = ConvBlock(channels, channels, kernel_size, padding=kernel_size // 2, activation='none') |
| 38 | self.activation = nn.ReLU(inplace=True) |
| 39 | |
| 40 | def forward(self, x): |
| 41 | residual = x |