Update docs

Signed-off-by: Beat Buesser <beat.buesser@ibm.com>
Trusted-AI · Dec 27, 2023 · 501ad92 · 501ad92
1 parent bc8a15f
commit 501ad92
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/docs/modules/attacks/evasion.rst b/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

diff --git a/docs/modules/defences/trainer.rst b/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