| 154 | |
| 155 | |
| 156 | class BasicConv2d(nn.Module): |
| 157 | def __init__(self, in_channels, out_channels, **kwargs): |
| 158 | super(BasicConv2d, self).__init__() |
| 159 | self.conv = nn.Conv2d(in_channels, out_channels, **kwargs) |
| 160 | self.relu = nn.ReLU(inplace=True) |
| 161 | |
| 162 | def forward(self, x): |
| 163 | x = self.conv(x) |
| 164 | x = self.relu(x) |
| 165 | return x |