    def train(self, mode: bool = True):
        """

        # Parameters

        mode : `bool`, optional (default=`True`)
            Sets `requires_grad` to value of `mode` for bias mitigator
            and associated bias direction.
        """
        raise NotImplementedError
# Parameters

mode : `bool`, optional (default=`True`)
Sets `requires_grad` to value of `mode` for bias mitigator
and associated bias direction.
allennlp/fairness/bias_mitigator_wrappers.py