Update docs

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

docs/modules/attacks/evasion.rst

@@ -86,6 +86,12 @@ Carlini and Wagner ASR Attack
    :members:
    :special-members:
 
+Composite Adversarial Attack - PyTorch
+--------------------------------------
+.. autoclass:: CompositeAdversarialAttackPyTorch
+   :members:
+   :special-members:
+
 Decision Tree Attack
 --------------------
 .. autoclass:: DecisionTreeAttack

docs/modules/defences/trainer.rst

@@ -20,6 +20,24 @@ Adversarial Training Madry PGD
    :members:
    :special-members:
 
+Adversarial Training Adversarial Weight Perturbation (AWP) - PyTorch
+--------------------------------------------------------------------
+.. autoclass:: AdversarialTrainerAWPPyTorch
+   :members:
+   :special-members:
+
+Adversarial Training Oracle Aligned Adversarial Training (OAAT) - PyTorch
+-------------------------------------------------------------------------
+.. autoclass:: AdversarialTrainerOAATPyTorch
+   :members:
+   :special-members:
+
+Adversarial Training TRADES - PyTorch
+-------------------------------------
+.. autoclass:: AdversarialTrainerTRADESPyTorch
+   :members:
+   :special-members:
+
 Base Class Adversarial Training Fast is Better than Free
 --------------------------------------------------------
 .. autoclass:: AdversarialTrainerFBF