def reset_classifier(self, n_outputs):
        """
        Reset the classifier with the new number of classes.

        Parameters
        ----------
        n_outputs : int
            The new number of classes.
        """
        self.n_outputs = n_outputs
        self.final_layer = (
            nn.Linear(self.emb_dim, self.n_outputs)
            if self.n_outputs > 0
            else nn.Identity()
        )

        Reset the classifier with the new number of classes.

        Parameters
        ----------
        n_outputs : int
            The new number of classes.
        
braindecode/models/labram.py