Method __init__

ML/src/python/neuralforge/nn/activations.py:41–44

(self, channels, kernel_size=3)

class FReLU(nn.Module):
    def __init__(self, channels, kernel_size=3):
        super().__init__()
        self.conv = nn.Conv2d(channels, channels, kernel_size, padding=kernel_size // 2, groups=channels)
        self.bn = nn.BatchNorm2d(channels)
    
    def forward(self, x):
        tx = self.bn(self.conv(x))