def forward(
        self,
        X: torch.Tensor,
    ) -> torch.Tensor:
        """
        Forward pass of the residual module.

        Parameters
        ----------
        X : torch.Tensor
            Input tensor of shape (batch_size, ch_names, n_times).

        Returns
        -------
        torch.Tensor
            Output tensor after convolution, batch normalization, and activation function.
        """
        out = self.conv(X)
        out = self.bn(out)
        return self.activation(out)

        Forward pass of the residual module.

        Parameters
        ----------
        X : torch.Tensor
            Input tensor of shape (batch_size, ch_names, n_times).

        Returns
        -------
        torch.Tensor
            Output tensor after convolution, batch normalization, and activation function.
        
braindecode/models/eeginception_mi.py