advsecurenet.defenses package

---

advsecurenet.defenses.adversarial_training module

class advsecurenet.defenses.adversarial_training.AdversarialTraining(config: AdversarialTrainingConfig)

Bases: Trainer

Adversarial Training class. This module implements the Adversarial Training defense.

Parameters:

config (AdversarialTrainingConfig) – The configuration for the Adversarial Training defense.

advsecurenet.defenses.ddp_adversarial_training module

class advsecurenet.defenses.ddp_adversarial_training.DDPAdversarialTraining(config: AdversarialTrainingConfig, rank: int, world_size: int)

Bases: DDPTrainer, AdversarialTraining

Adversarial Training class. This class is used to train a model using adversarial training.

Note

This module inherits from both DDPTrainer and AdversarialTraining. The order of inheritance is important because of MRO.

---