def __init__(
        self,
        classifier: "CLASSIFIER_LOSS_GRADIENTS_TYPE",
        attack: EvasionAttack,
        beta: float = 6.0,
    ):
        """
        Create an :class:`.AdversarialTrainerTRADES` instance.

        :param classifier: Model to train adversarially.
        :param attack: attack to use for data augmentation in adversarial training
        :param beta: The scaling factor controlling tradeoff between clean loss and adversarial loss
        """
        self._attack = attack
        self._beta = beta
        super().__init__(classifier)

        Create an :class:`.AdversarialTrainerTRADES` instance.

        :param classifier: Model to train adversarially.
        :param attack: attack to use for data augmentation in adversarial training
        :param beta: The scaling factor controlling tradeoff between clean loss and adversarial loss
        
art/defences/trainer/adversarial_trainer_trades.py