def METHOD_NAME(self, in_channels, stem_channels):
    self.conv1 = nn.Conv2d(
        in_channels, stem_channels, kernel_size=3, stride=1, padding=1)
    self.bn1 = nn.BatchNorm2d(stem_channels)
    self.relu1 = nn.ReLU(inplace=True)