def __init__(self, activation: nn.Module = nn.GELU):
        """

        Parameters
        ----------
        activation: nn.Module
            Activation function for the InterFrequency Module

        """
        super().__init__()

        self.activation = activation()


        Parameters
        ----------
        activation: nn.Module
            Activation function for the InterFrequency Module

        
braindecode/models/ifnet.py